wchliao/multi-task-image-classification
Multi-task learning for image classification implemented in PyTorch.
Python
Issues
- 1
cuda out of memory even batch_size = 2
#2 opened by Light-- - 2
network architecture
#1 opened by Light--
Multi-task learning for image classification implemented in PyTorch.
Python