Update app.py
Browse files
app.py
CHANGED
|
@@ -56,12 +56,12 @@ def tryon(person,cloth,edge):
|
|
| 56 |
warp_model = AFWM(opt, 3)
|
| 57 |
print(warp_model)
|
| 58 |
warp_model.eval()
|
| 59 |
-
warp_model.cuda()
|
| 60 |
load_checkpoint(warp_model, opt.warp_checkpoint)
|
| 61 |
|
| 62 |
gen_model = ResUnetGenerator(7, 4, 5, ngf=64, norm_layer=nn.BatchNorm2d)
|
| 63 |
gen_model.eval()
|
| 64 |
-
gen_model.cuda()
|
| 65 |
load_checkpoint(gen_model, opt.gen_checkpoint)
|
| 66 |
|
| 67 |
total_steps = (start_epoch-1) * dataset_size + epoch_iter
|
|
|
|
| 56 |
warp_model = AFWM(opt, 3)
|
| 57 |
print(warp_model)
|
| 58 |
warp_model.eval()
|
| 59 |
+
#warp_model.cuda()
|
| 60 |
load_checkpoint(warp_model, opt.warp_checkpoint)
|
| 61 |
|
| 62 |
gen_model = ResUnetGenerator(7, 4, 5, ngf=64, norm_layer=nn.BatchNorm2d)
|
| 63 |
gen_model.eval()
|
| 64 |
+
#gen_model.cuda()
|
| 65 |
load_checkpoint(gen_model, opt.gen_checkpoint)
|
| 66 |
|
| 67 |
total_steps = (start_epoch-1) * dataset_size + epoch_iter
|