From 66676eb03939fe0573f584022a427da948cd8a83 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 6 Oct 2020 15:00:47 +0200 Subject: [PATCH] init_torch_seeds >> init_seeds bug fix --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 7987537..c6fa311 100644 --- a/train.py +++ b/train.py @@ -26,9 +26,9 @@ from utils.datasets import create_dataloader from utils.general import ( torch_distributed_zero_first, labels_to_class_weights, plot_labels, check_anchors, labels_to_image_weights, compute_loss, plot_images, fitness, strip_optimizer, plot_results, get_latest_run, check_dataset, check_file, - check_git_status, check_img_size, increment_dir, print_mutation, plot_evolution, set_logging) + check_git_status, check_img_size, increment_dir, print_mutation, plot_evolution, set_logging, init_seeds) from utils.google_utils import attempt_download -from utils.torch_utils import init_seeds, ModelEMA, select_device, intersect_dicts +from utils.torch_utils import ModelEMA, select_device, intersect_dicts logger = logging.getLogger(__name__)