Update train.py

Author: bubbliiiing

train.py

@@ -160,9 +160,9 @@ def fit_ont_epoch(net,yolo_losses,epoch,epoch_size,epoch_size_val,gen,genval,Epo
         if Use_Data_Loader:
             train_dataset = YoloDataset(lines[:num_train], (Config["img_h"], Config["img_w"]))
             val_dataset = YoloDataset(lines[num_train:], (Config["img_h"], Config["img_w"]))
-            gen = DataLoader(train_dataset, batch_size=Batch_size, num_workers=0, pin_memory=True,
+            gen = DataLoader(train_dataset, batch_size=Batch_size, num_workers=4, pin_memory=True,
                                     drop_last=True, collate_fn=yolo_dataset_collate)
-            gen_val = DataLoader(val_dataset, batch_size=Batch_size, num_workers=0,pin_memory=True, 
+            gen_val = DataLoader(val_dataset, batch_size=Batch_size, num_workers=4,pin_memory=True, 
                                     drop_last=True, collate_fn=yolo_dataset_collate)
         else:
             gen = Generator(Batch_size, lines[:num_train],
@@ -193,9 +193,9 @@ def fit_ont_epoch(net,yolo_losses,epoch,epoch_size,epoch_size_val,gen,genval,Epo
         if Use_Data_Loader:
             train_dataset = YoloDataset(lines[:num_train], (Config["img_h"], Config["img_w"]))
             val_dataset = YoloDataset(lines[num_train:], (Config["img_h"], Config["img_w"]))
-            gen = DataLoader(train_dataset, batch_size=Batch_size, num_workers=0, pin_memory=True,
+            gen = DataLoader(train_dataset, batch_size=Batch_size, num_workers=4, pin_memory=True,
                                     drop_last=True, collate_fn=yolo_dataset_collate)
-            gen_val = DataLoader(val_dataset, batch_size=Batch_size, num_workers=0,pin_memory=True, 
+            gen_val = DataLoader(val_dataset, batch_size=Batch_size, num_workers=4,pin_memory=True, 
                                     drop_last=True, collate_fn=yolo_dataset_collate)
         else:
             gen = Generator(Batch_size, lines[:num_train],