Thank you very much for your research and for sharing the code.
When I attempted to replicate your model using the following code:
model = get_ric_vgg(BATCH_SIZE=128)
model = model.cuda()
x = torch.randn(4, 3, 28, 28).cuda()
out = model(x)
print(out)
I encountered a RuntimeError: invalid batch size of offset.
Could you please assist me in identifying any configurations that I may have overlooked or not set correctly? I would greatly appreciate your response.
Hi!
Thank you very much for your research and for sharing the code.
When I attempted to replicate your model using the following code:
I encountered a RuntimeError: invalid batch size of offset.
Could you please assist me in identifying any configurations that I may have overlooked or not set correctly? I would greatly appreciate your response.
Thank you!