Hello, thanks for this project! I'm trying to add a training code to mlp-mixer using the CIFAR dataset. I have added a transform then to adapt the images:
for i, data in enumerate(trainloader, 0):
inputs, labels = data
inputs, labels = inputs.to(device), labels.to(device)
print("Input:", inputs.shape)
outputs = mixer_model(inputs)
print(outputs)
if i == 1:
break
Hello, thanks for this project! I'm trying to add a training code to
mlp-mixer
using the CIFAR dataset. I have added atransform
then to adapt the images:and to try out image representation in the model:
and I get
while a naive training code, I'm not actually sure if the input image resize to the model is correct. Thank you.