Shortcuts

OutputTransform

class flash.core.data.io.output_transform.OutputTransform[source]

The OutputTransform encapsulates all the data processing logic that should run after the model.

static per_batch_transform(batch)[source]

Transforms to apply on a whole batch before uncollation to individual samples.

Can involve both CPU and Device transforms as this is not applied in separate workers.

Return type

Any

static per_sample_transform(sample)[source]

Transforms to apply to a single sample after splitting up the batch.

Can involve both CPU and Device transforms as this is not applied in separate workers.

Return type

Any

static uncollate(batch)[source]

Uncollates a batch into single samples.

Tries to preserve the type wherever possible.

Return type

Any

Read the Docs v: latest
Versions
latest
stable
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.2
0.5.1
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
docs-fix_typing
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.