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