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

More working than not

Browse source 
Not sure if validation works, call it a day
This commit is contained in:
Zhengyi Chen 2024-03-03 03:16:54 +00:00
parent 4a03211c83
commit 12aabb0d3f
10 changed files with 116 additions and 105 deletions
Download patch file Download diff file Expand all files Collapse all files

4
arguments.py
View file

@ -12,13 +12,13 @@ parser.add_argument(
# Data configuration =========================================================
parser.add_argument(
"--worker", type=int, default=4, help="Number of data loader processes"
"--workers", type=int, default=4, help="Number of data loader processes"
)
parser.add_argument(
"--train_dataset", type=str, default="ShanghaiA", help="Training dataset"
)
parser.add_argument(
"--test_dataset", type=str, default="ShanghaiA", help="Evaluation dataset"
"--eval_dataset", type=str, default="ShanghaiA", help="Evaluation dataset"
)
parser.add_argument(
"--print_freq", type=int, default=1,