From 99205d7be54abebdd7ef39f7c9887bba65906ba1 Mon Sep 17 00:00:00 2001 From: Jakub Kaczmarzyk Date: Wed, 25 May 2022 19:10:34 -0400 Subject: [PATCH] Revert "ENH: replace star imports with imported names in train.py" --- train.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/train.py b/train.py index acdf93c3..c953eb02 100755 --- a/train.py +++ b/train.py @@ -31,11 +31,8 @@ from torch.nn.parallel import DistributedDataParallel as NativeDDP from timm.data import create_dataset, create_loader, resolve_data_config, Mixup, FastCollateMixup, AugMixDataset from timm.models import create_model, safe_model_name, resume_checkpoint, load_checkpoint,\ convert_splitbn_model, model_parameters -from timm.utils import setup_default_logging, random_seed, set_jit_fuser, ModelEmaV2,\ - get_outdir, CheckpointSaver, distribute_bn, update_summary, accuracy, AverageMeter,\ - dispatch_clip_grad, reduce_tensor -from timm.loss import JsdCrossEntropy, BinaryCrossEntropy, SoftTargetCrossEntropy, BinaryCrossEntropy,\ - LabelSmoothingCrossEntropy +from timm.utils import * +from timm.loss import * from timm.optim import create_optimizer_v2, optimizer_kwargs from timm.scheduler import create_scheduler from timm.utils import ApexScaler, NativeScaler