AdalAbilbekov commited on
Commit
4b2875e
1 Parent(s): ef42a25

I hope this one is final commit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ model = SpecClassifier(
42
  ckpt = './cnnwt_SGD_1959.pt'
43
  ckpt_tts = './grad_uncond_cnn_001.pt'
44
 
45
- utils.load_checkpoints_no_logger(ckpt_tts, gradtts_uncond_model, None)
46
- utils.load_checkpoints_no_logger(ckpt, model, None)
47
 
48
  _ = model.to(device).eval()
49
 
 
42
  ckpt = './cnnwt_SGD_1959.pt'
43
  ckpt_tts = './grad_uncond_cnn_001.pt'
44
 
45
+ utils.load_checkpoint_no_logger(ckpt_tts, gradtts_uncond_model, None)
46
+ utils.load_checkpoint_no_logger(ckpt, model, None)
47
 
48
  _ = model.to(device).eval()
49