Channel first or last?

Perhaps it is a stupid question but… Which is the default channel ordering in fast.ai: first or last? Is it possible to choose depending on your needs?

For images, we follow PyTorch convention: batch size, channel, height, width.
For text, the order is batch size, sequence length.

2 Likes