multi-task-image-classification icon indicating copy to clipboard operation
multi-task-image-classification copied to clipboard

Published 20 hours ago • verified publisher icon wchliao
→

Metadata

Multi-task learning for image classification implemented in PyTorch.

Results 1 multi-task-image-classification issues
Sort by recently updated
recently updated
newest added

cuda out of memory even batch_size = 2

1 comment

hi, i modified the input size to receive 112x112 and fc layer, but always get out of memory after running.... The locations I modified in the code are as follows:...

Light-- avatar Light--

← Metadata

95
Stars
28
Forks
Watchers

Owner

verified publisher icon wchliao

Metadata

Multi-task learning for image classification implemented in PyTorch.

Back