Not able to train on imagenet using the inception architecture

vision
7

I think the actual code here would be more helpful than just a notebook link.

if isinstance(outputs, tuple):
    loss = sum((criterion(o,labels) for o in outputs))
else:
    loss = criterion(outputs, labels)
8 Likes