Spaces:
Runtime error
Runtime error
邱浩楠
commited on
Commit
•
9cfe57b
1
Parent(s):
3238866
init
Browse files- models/sample_model.py +1 -1
- options/test_options.py +1 -1
models/sample_model.py
CHANGED
@@ -52,7 +52,7 @@ class SampleModel(BaseModel):
|
|
52 |
By default, we use vanilla GAN loss, UNet with batchnorm, and aligned datasets.
|
53 |
"""
|
54 |
# changing the default values to match the pix2pix paper (https://phillipi.github.io/pix2pix/)
|
55 |
-
parser.set_defaults(norm='batch', netG='unet_256', dataset_mode='noise', num_test = 32
|
56 |
parser.add_argument('--pose_path', type=str, default='', help='path for pose net')
|
57 |
parser.add_argument('--rnn_path', type=str, default='', help='path for rnn net')
|
58 |
parser.add_argument('--n_frames_G', type=int, default=60)
|
|
|
52 |
By default, we use vanilla GAN loss, UNet with batchnorm, and aligned datasets.
|
53 |
"""
|
54 |
# changing the default values to match the pix2pix paper (https://phillipi.github.io/pix2pix/)
|
55 |
+
parser.set_defaults(norm='batch', netG='unet_256', dataset_mode='noise', num_test = 32)
|
56 |
parser.add_argument('--pose_path', type=str, default='', help='path for pose net')
|
57 |
parser.add_argument('--rnn_path', type=str, default='', help='path for rnn net')
|
58 |
parser.add_argument('--n_frames_G', type=int, default=60)
|
options/test_options.py
CHANGED
@@ -16,7 +16,7 @@ class TestOptions(BaseOptions):
|
|
16 |
parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
|
17 |
parser.add_argument('--num_test', type=int, default=50, help='how many test images to run')
|
18 |
# rewrite devalue values
|
19 |
-
parser.set_defaults(model='test')
|
20 |
# To avoid cropping, the load_size should be the same as crop_size
|
21 |
parser.set_defaults(load_size=parser.get_default('crop_size'))
|
22 |
self.isTrain = False
|
|
|
16 |
parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
|
17 |
parser.add_argument('--num_test', type=int, default=50, help='how many test images to run')
|
18 |
# rewrite devalue values
|
19 |
+
# parser.set_defaults(model='test')
|
20 |
# To avoid cropping, the load_size should be the same as crop_size
|
21 |
parser.set_defaults(load_size=parser.get_default('crop_size'))
|
22 |
self.isTrain = False
|