Revert "ENH: replace star imports with imported names in train.py"

pull/1273/head
Jakub Kaczmarzyk 3 years ago committed by GitHub
parent e4360e6125
commit 99205d7be5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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.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,\ from timm.models import create_model, safe_model_name, resume_checkpoint, load_checkpoint,\
convert_splitbn_model, model_parameters convert_splitbn_model, model_parameters
from timm.utils import setup_default_logging, random_seed, set_jit_fuser, ModelEmaV2,\ from timm.utils import *
get_outdir, CheckpointSaver, distribute_bn, update_summary, accuracy, AverageMeter,\ from timm.loss import *
dispatch_clip_grad, reduce_tensor
from timm.loss import JsdCrossEntropy, BinaryCrossEntropy, SoftTargetCrossEntropy, BinaryCrossEntropy,\
LabelSmoothingCrossEntropy
from timm.optim import create_optimizer_v2, optimizer_kwargs from timm.optim import create_optimizer_v2, optimizer_kwargs
from timm.scheduler import create_scheduler from timm.scheduler import create_scheduler
from timm.utils import ApexScaler, NativeScaler from timm.utils import ApexScaler, NativeScaler

Loading…
Cancel
Save