Update models for multiclass

[daniprec]

Update our pytorch models to perform multiclass classification.
See cross entropy loss.

[daniprec]

⚠️ Notice: This project is no longer actively maintained, and this issue won't be addressed. Feel free to continue the discussion, but be aware that there won't be further updates.