batch_size (int
) – batch size
img_size (Union
[int
, Tuple
[int
, int
]]) – height and width. Default: 224
channels (int
) – color channels. Default: 3
length (int
) – loader length. Default: 100
num_classes (int
) – number of classes. Default: 1000