Saved_Models_Weights/tinyimagenet/tinyimagenet_62.51.txt

model = WaveMix(
    num_classes = 200,
    depth = 12,
    mult = 2,
    ff_channel = 256,
    final_dim = 256,
    dropout = 0.5,
    level=1,
    initial_conv = 'strided',
    stride = 1

)