could you please share me the configurations of packeges ? I trired with following configurations: python3.6, pytorch 1.2.0, torchvision 0.4.0, cuda 10.0, gcc4.9 in ubuntu-20
But i am encountering by the errors :
##############
class GradScaler(torch.cuda.amp.GradScaler):
AttributeError: module 'torch.cuda' has no attribute 'amp'
hi, @yashasbharadwaj111 I'm sorry. I can't share our model because our data set is divided into two parts:
Originally posted by @AaronWong in https://github.com/NVlabs/few-shot-vid2vid/issues/14#issuecomment-570766853
could you please share me the configurations of packeges ? I trired with following configurations: python3.6, pytorch 1.2.0, torchvision 0.4.0, cuda 10.0, gcc4.9 in ubuntu-20 But i am encountering by the errors : ############## class GradScaler(torch.cuda.amp.GradScaler): AttributeError: module 'torch.cuda' has no attribute 'amp'