diff --git a/train_with_tensorboard.py b/train_with_tensorboard.py index 0eecfc8e2dc58590ba254a2552af96d3317015e0..400c74c00ef4c66e50ea306b918897fd58766de9 100644 --- a/train_with_tensorboard.py +++ b/train_with_tensorboard.py @@ -202,7 +202,7 @@ if __name__ == "__main__": if Cosine_lr: lr_scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=5, eta_min=1e-5) else: - lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.9) + lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95) if Use_Data_Loader: train_dataset = YoloDataset(lines[:num_train], (input_shape[0], input_shape[1]), mosaic=mosaic) @@ -239,7 +239,7 @@ if __name__ == "__main__": if Cosine_lr: lr_scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=5, eta_min=1e-5) else: - lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.9) + lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95) if Use_Data_Loader: train_dataset = YoloDataset(lines[:num_train], (input_shape[0], input_shape[1]), mosaic=mosaic)