IDK, it keeps going over 32.2GB then it gives out of memory, on SPDA. I'll keep trying other thin
IDK, it keeps going over 32.2GB then it gives out of memory, on SPDA. I'll keep trying other things.


torch.cpu.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cpu', args...) instead.torch.cpu.amp.autocast(args...)torch.amp.autocast('cpu', args...)