Target has to be one element

MathGaron / pytorch_toolbox

Boiler plate code for pytorch. Train/Validation loops, visualization etc. For research.

MIT License

10 stars 3 forks source link

Target has to be one element #6

Closed jacenfox closed 7 years ago

jacenfox commented 7 years ago

https://github.com/MathGaron/pytorch_toolbox/blob/f06b1894f3a140792a31af61cbdd03b64b3cae4e/pytorch_toolbox/train_loop.py#L181 If target contains two items, for example, label+image, it goes wrong here.

jacenfox commented 7 years ago

I figured out how does it work. It should be okay!

MathGaron commented 7 years ago

Thats a good point though, it was a particular case implemented by aditya in our first version, and seriously I dont remember why he did that, but I think we should remove it.. I will do it later

MathGaron commented 7 years ago

will be resolved in pull request #11 so I close now