apply_channels_last

composer.functional.apply_channels_last(model)

Changes the memory format of the model to torch.channels_last.

This usually yields improved GPU utilization.

Parameters

model (Module) – The model or module to modify.