Skip to content

GitLab

Unverified Commit 601b568f authored by Maksim Skobeev's avatar Maksim Skobeev Committed by GitHub
Browse files

Update model.py

parent a9a671cb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
RealESRGAN/model.py
View file @ 601b568f
...@@ -56,9 +56,10 @@ class RealESRGAN: ...@@ -56,9 +56,10 @@ class RealESRGAN:
self.model.eval() self.model.eval()
self.model.to(self.device) self.model.to(self.device)
@torch.cuda.amp.autocast() # @torch.cuda.amp.autocast()
def predict(self, lr_image, batch_size=4, patches_size=192, def predict(self, lr_image, batch_size=4, patches_size=192,
padding=24, pad_size=15): padding=24, pad_size=15):
torch.autocast(device_type=self.device.type)
scale = self.scale scale = self.scale
device = self.device device = self.device
lr_image = np.array(lr_image) lr_image = np.array(lr_image)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment