Thank you for the code.



 hyunwookim:

I did not include this part because it takes inputs from my custom made dataloaders (a feature which I excluded from the minimal model), and other than that all it does is loop (but I should have, since I just noticed the optimizer is here). Besides, are you going to make minimal inputs yourself or do you need the images that I used for the model?


If you could tell the shape of your inputs, we can construct random tensors to match that shape, and that is usually enough to flush out many bugs.
Edited to add: Wait, where is the code that invokes run_model? Perhaps the bug is in the way you construct model?