unilab.ipc.replay_pipelines.base.ReplayPipeline

class unilab.ipc.replay_pipelines.base.ReplayPipeline[source]

Bases: Protocol

Methods

__init__(*args, **kwargs)

after_tick()

batch_ready(tick_id, sample_count)

close()

sample_large_batch(tick_id, sample_count)

start_prepare(tick_id, sample_count[, ...])

wait_ready()

wait_until_ready(tick_id, sample_count)

start_prepare(tick_id, sample_count, min_snapshot_ptr=None)[source]
Parameters:
Return type:

bool

batch_ready(tick_id, sample_count)[source]
Parameters:

  • tick_id (int)

  • sample_count (int)

Return type:

bool

wait_ready()[source]
Return type:

None

wait_until_ready(tick_id, sample_count)[source]
Parameters:

  • tick_id (int)

  • sample_count (int)

Return type:

bool

sample_large_batch(tick_id, sample_count)[source]
Parameters:

  • tick_id (int)

  • sample_count (int)

Return type:

Dict[str, Tensor]

after_tick()[source]
Return type:

None

close()[source]
Return type:

None

__init__(*args, **kwargs)