update multi gpus

bubbliiiing · bubbliiiing · commit 5524b5e54504 · 2022-04-14T18:32:14.000+08:00
diff --git a/train.py b/train.py
@@ -460,4 +460,4 @@
 
             set_optimizer_lr(optimizer, lr_scheduler_func, epoch)
 
-            fit_one_epoch(model_train, model, yolo_loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, UnFreeze_Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank, local_rank)
+            fit_one_epoch(model_train, model, yolo_loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, UnFreeze_Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank)

Original file line number	Diff line number	Diff line change
`@@ -460,4 +460,4 @@`
`460`	`460`
`461`	`461`	`set_optimizer_lr(optimizer, lr_scheduler_func, epoch)`
`462`	`462`
`463`		`- fit_one_epoch(model_train, model, yolo_loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, UnFreeze_Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank, local_rank)`
	`463`	`+ fit_one_epoch(model_train, model, yolo_loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, UnFreeze_Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank)`