mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

rm whitespace

Browse Source
This commit is contained in:
Jakub Kaczmarzyk 2022-05-18 10:17:02 -04:00
parent 20a1fa63f8
commit e1e4c9bbae
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -55,7 +55,7 @@ except AttributeError:
try: try:
import wandb import wandb
has_wandb = True has_wandb = True
except ImportError: except ImportError:
has_wandb = False has_wandb = False
torch.backends.cudnn.benchmark = True torch.backends.cudnn.benchmark = True
@ -326,14 +326,14 @@ def _parse_args():
def main(): def main():
setup_default_logging() setup_default_logging()
args, args_text = _parse_args() args, args_text = _parse_args()
if args.log_wandb: if args.log_wandb:
if has_wandb: if has_wandb:
wandb.init(project=args.experiment, config=args) wandb.init(project=args.experiment, config=args)
else: else:
_logger.warning("You've requested to log metrics to wandb but package not found. " _logger.warning("You've requested to log metrics to wandb but package not found. "
"Metrics not being logged to wandb, try `pip install wandb`") "Metrics not being logged to wandb, try `pip install wandb`")
args.prefetcher = not args.no_prefetcher args.prefetcher = not args.no_prefetcher
args.distributed = False args.distributed = False
if 'WORLD_SIZE' in os.environ: if 'WORLD_SIZE' in os.environ: