Commit 64582fb9 authored by Gabriele Franch's avatar Gabriele Franch
Browse files

restored patch size from command line args

parent 1c86e437
......@@ -38,7 +38,7 @@ parser.add_argument("--pretrained_model", type=str, default="")
parser.add_argument("--num_hidden", type=str, default="64,64,64,64")
parser.add_argument("--filter_size", type=int, default=5)
parser.add_argument("--stride", type=int, default=1)
# parser.add_argument("--patch_size", type=int, default=4)
parser.add_argument("--patch_size", type=int, default=4)
parser.add_argument("--layer_norm", type=int, default=1)
# scheduled sampling
......@@ -60,12 +60,12 @@ parser.add_argument("--num_save_samples", type=int, default=10)
args = parser.parse_args()
batch_size = args.batch_size
if args.img_width == 128:
args.patch_size = 4
elif args.img_width == 256:
args.patch_size = 8
elif args.img_width == 512:
args.patch_size = 16
# if args.img_width == 128:
# args.patch_size = 4
# elif args.img_width == 256:
# args.patch_size = 8
# elif args.img_width == 512:
# args.patch_size = 16
def schedule_sampling(eta: float, itr: int):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment