from wavemix.classification import WaveMix

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

)