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

Yay, works in DP via CPU

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-03 21:59:58 +00:00
parent ab15419d2f
commit fc941ebaf7
6 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
arguments.py
View file

@ -76,7 +76,7 @@ parser.add_argument(
"--ddp_world_size", type=int, default=1,
help="DDP: Number of processes in Pytorch process group"
)
parse.add_argument(
parser.add_argument(
"--debug", type=bool, default=False
)