hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

para modified

Browse Source
This commit is contained in:
chun 2023-12-23 14:47:20 +08:00
parent 6d91d72581
commit 05cdfb29c0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -33,7 +33,7 @@ def config_parser():
parser.add_argument('--num_workers', default=0, type=int, help='num_workers') parser.add_argument('--num_workers', default=0, type=int, help='num_workers')
parser.add_argument('--dataset', default='cifar10', type=str, parser.add_argument('--dataset', default='cifar10', type=str,
choices=['cifar10', 'imagenet'], help='dataset') choices=['cifar10', 'imagenet'], help='dataset')
parser.add_argument('--parallel', default=True, type=bool, help='parallel') parser.add_argument('--parallel', default=False, type=bool, help='parallel')
return parser.parse_args() return parser.parse_args()