Spaces:
Running
Running
height
Browse files
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 =
|
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
|