Skip to content

Commit fd360ac

Browse files
authored
Merge pull request #1266 from kaczmarj/enh/no-star-imports
ENH: replace star imports with imported names in train.py
2 parents 004fb5f + ce5578b commit fd360ac

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

train.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,11 @@
3131
from timm.data import create_dataset, create_loader, resolve_data_config, Mixup, FastCollateMixup, AugMixDataset
3232
from timm.models import create_model, safe_model_name, resume_checkpoint, load_checkpoint,\
3333
convert_splitbn_model, model_parameters
34-
from timm.utils import *
35-
from timm.loss import *
34+
from timm.utils import setup_default_logging, random_seed, set_jit_fuser, ModelEmaV2,\
35+
get_outdir, CheckpointSaver, distribute_bn, update_summary, accuracy, AverageMeter,\
36+
dispatch_clip_grad, reduce_tensor
37+
from timm.loss import JsdCrossEntropy, BinaryCrossEntropy, SoftTargetCrossEntropy, BinaryCrossEntropy,\
38+
LabelSmoothingCrossEntropy
3639
from timm.optim import create_optimizer_v2, optimizer_kwargs
3740
from timm.scheduler import create_scheduler
3841
from timm.utils import ApexScaler, NativeScaler

0 commit comments

Comments
 (0)