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.8.2
0.8.1.post0
0.8.1
0.8.0
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
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.