search

ai-forever / Real-ESRGAN

PyTorch implementation of Real-ESRGAN model
BSD 3-Clause "New" or "Revised" License
463 stars 118 forks source link

Update model.py #14

Open nunombispo opened 1 year ago

nunombispo commented 1 year ago

Replace @torch.cuda.amp.autocast() with torch.autocast(device_type=self.device.type) for proper device recognition