alrao

alrao copied to clipboard

memory consumption too high

1

Hi First of all, I really liked the paper and the implementation. but when i ran it for VGG19 network, it took almost 10.5GB of GPU RAM( I have 1...

aakash-saboo

Is the code written only for nn.NLLL() loss?

3

I changed the loss to MultLabelSoftMargin classifier. My input is a regular 4 dimension image array and targets are one hot encoded vectors(with vector length = n_classes). I am performing...

aakash-saboo