gyrojeff commited on
Commit
65e514f
1 Parent(s): 69fba53

feat: add windows num_worker flag

Browse files
Files changed (1) hide show
  1. train.py +2 -0
train.py CHANGED
@@ -86,6 +86,8 @@ single_batch_size = args.single_batch_size
86
  total_num_workers = os.cpu_count()
87
  single_device_num_workers = total_num_workers // len(devices)
88
 
 
 
89
 
90
  lr = args.lr
91
  b1 = 0.9
 
86
  total_num_workers = os.cpu_count()
87
  single_device_num_workers = total_num_workers // len(devices)
88
 
89
+ if os.name == "nt":
90
+ single_device_num_workers = 0
91
 
92
  lr = args.lr
93
  b1 = 0.9