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

Debug

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-03 19:40:22 +00:00
parent a9dd8dee04
commit ab15419d2f
5 changed files with 63 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

5
arguments.py
View file

@ -63,7 +63,7 @@ parser.add_argument(
"--epochs", type=int, default=250, help="Number of epochs to train"
)
parser.add_argument(
"--gpus", type=List[int], default=[0],
"--gpus", type=str, default='0',
help="GPU IDs to be made available for training runtime"
)
@ -76,6 +76,9 @@ parser.add_argument(
"--ddp_world_size", type=int, default=1,
help="DDP: Number of processes in Pytorch process group"
)
parse.add_argument(
"--debug", type=bool, default=False
)
# nni configuration ==========================================================
parser.add_argument(