As above, I keep getting the error ‘Target 2 is out of bounds’ when trying to train a MetaCAT model with 3 possible outputs. From this thread, IndexError: Target 2 is out of bounds - #12 by Vijaya_kumar - vision - PyTorch Forums, I believe it is to do with the input tensor being the wrong size but I’m struggling to debug it looking at the git hub code. This happens when using the model.train_raw and model.train_from_json functions. Any thoughts?
Thanks