Spaces:
Runtime error
Runtime error
Remove optimization :(
Browse files
test.py
CHANGED
@@ -33,7 +33,6 @@ img = img * 1.0 / 255
|
|
33 |
img = torch.from_numpy(img[np.newaxis, :, :]).float()
|
34 |
img_LR = img.unsqueeze(0)
|
35 |
img_LR = img_LR.to(device)
|
36 |
-
img_LR = img_LR.pin_memory()
|
37 |
|
38 |
print('Start upscaling...')
|
39 |
with torch.no_grad():
|
|
|
33 |
img = torch.from_numpy(img[np.newaxis, :, :]).float()
|
34 |
img_LR = img.unsqueeze(0)
|
35 |
img_LR = img_LR.to(device)
|
|
|
36 |
|
37 |
print('Start upscaling...')
|
38 |
with torch.no_grad():
|