Mehdi Cherti commited on
Commit
a4a6a13
1 Parent(s): 6c1d070
Files changed (1) hide show
  1. test_ddgan.py +2 -2
test_ddgan.py CHANGED
@@ -425,11 +425,11 @@ def sample_and_test(args):
425
  args.epoch_id = epoch
426
 
427
  path = './saved_info/dd_gan/{}/{}/netG_{}.pth'.format(cfg['dataset'], args.name, args.epoch_id)
428
- next_next_path = './saved_info/dd_gan/{}/{}/netG_{}.pth'.format(cfg['dataset'], args.name, args.epoch_id+2)
429
  print(path)
430
  if not os.path.exists(path):
431
  continue
432
- if not os.path.exists(next_next_path):
433
  break
434
  print("PATH", path)
435
  suffix = '_' + args.eval_name if args.eval_name else ""
 
425
  args.epoch_id = epoch
426
 
427
  path = './saved_info/dd_gan/{}/{}/netG_{}.pth'.format(cfg['dataset'], args.name, args.epoch_id)
428
+ next_path = './saved_info/dd_gan/{}/{}/netG_{}.pth'.format(cfg['dataset'], args.name, args.epoch_id+1)
429
  print(path)
430
  if not os.path.exists(path):
431
  continue
432
+ if not os.path.exists(next_path):
433
  break
434
  print("PATH", path)
435
  suffix = '_' + args.eval_name if args.eval_name else ""