NVlabs/DeepInversion

AttributeError: module 'torch.cuda.amp' has no attribute 'initialize'

Egg-Hu opened this issue · 2 comments

Egg-Hu commented

AttributeError: module 'torch.cuda.amp' has no attribute 'initialize'

xmutly commented

I met same error, maybe the vesion of your torch is too new.