zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Test

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-02 17:52:10 +00:00
parent 6456d3878e
commit 359ed8c579
2 changed files with 29 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

16
arguments.py
View file

@ -11,6 +11,9 @@ parser.add_argument(
)
# Data configuration =========================================================
parser.add_argument(
"--worker", type=int, default=4, help="Number of data loader processes"
)
parser.add_argument(
"--train_dataset", type=str, default="ShanghaiA", help="Training dataset"
)
@ -28,19 +31,16 @@ parser.add_argument(
"--save_path", type=str, default="./save/default/",
help="Directory to save checkpoints in"
)
parser.add_argument(
"--resume", type=str, default=None, help="Path to checkpoint-ed model pth"
)
# Model configuration ========================================================
parser.add_argument(
"--load_revnet_from", type=str, default=None,
help="Pre-trained reverse perspective model path"
"--model", type=str, default="base", help="Model variant: <base|stn>"
)
parser.add_argument(
"--load_csrnet_from", type=str, default=None,
help="Pre-trained CSRNet model path"
)
parser.add_argument(
"--load_transcrowd_from", type=str, default=None,
help="Pre-trained TransCrowd model path"
"--pth_tar", type=str, default=None, help="Path to pre-training model pth"
)
# Optimizer configuration ====================================================