debisoft commited on
Commit
91fd505
·
1 Parent(s): 5c22ded
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ beta2 = 0.02
106
  device = torch.device("cuda:0" if torch.cuda.is_available() else torch.device('cpu'))
107
  n_feat = 64 # 64 hidden dimension feature
108
  n_cfeat = 5 # context vector is of size 5
109
- height = 16 # 16x16 image
110
  save_dir = './weights/'
111
 
112
  # training hyperparameters
 
106
  device = torch.device("cuda:0" if torch.cuda.is_available() else torch.device('cpu'))
107
  n_feat = 64 # 64 hidden dimension feature
108
  n_cfeat = 5 # context vector is of size 5
109
+ height = 64 # 16x16 image
110
  save_dir = './weights/'
111
 
112
  # training hyperparameters