lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update train.py.

Browse Source
This commit is contained in:
Brainway
2022-11-01 02:44:12 +00:00
committed by Gitee
parent b461f8a2d4
commit 4e640531da
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -292,7 +292,7 @@ def main():
parser.add_argument("--pretrained_model", type=str, default=None, help="load pretrained model") #None
parser.add_argument("--output_dir", default="./output", type=str,
help="The output directory where checkpoints will be written.")
parser.add_argument("--img_size", default=320, type=int, help="Resolution size")
parser.add_argument("--img_size", default=600, type=int, help="Resolution size")
parser.add_argument("--train_batch_size", default=8, type=int,
help="Total batch size for training.")
parser.add_argument("--eval_batch_size", default=8, type=int,