train_2020-04-08-19-38-36_MobilenetV3_block_6.log
1.33 KB
2020-04-08-19-38-36
use seed 283
use dataset : E:/code/detection/data/Fifth_data/All
{'task': 'All/train_2020-04-08-19-38-36_model=MobilenetV3-ep=3000-block=6-class=8', 'modelname': 'MobilenetV3', 'output': 'output', 'checkpoint': 'output/All/14835_model=MobilenetV3-ep=3000-block=6/model_best.pth.tar', 'gpu': [2], 'data': {'train': 'E:/code/detection/data/Fifth_data/All', 'val': '../data/Fifth_data/All', 'test': '../data/Fifth_data/All'}, 'train': {'epochs': 3000, 'start-epoch': 0, 'batch-size': 256, 'worker': 16, 'resume': '', 'augment': True, 'size': 224, 'confidence': False, 'weight': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]}, 'predict': {'batch-size': 256, 'worker': 16, 'cam': False, 'normalize': True, 'save': False}, 'optimizer': {'type': 'SGD', 'lr': 0.001, 'momentum': 0.9, 'weight_decay': 0.0001}, 'loss': {'gamma': 2.0, 'alpha': 0.8}, 'model': {'blocks': 6, 'class': 8}, 'etc': {'tensorboard': False, 'print_freq': 10}, 'id': 'train_2020-04-08-19-38-36'}
Fatal error in main loop
Traceback (most recent call last):
File "E:\code\detection\trainer\train.py", line 104, in main
run_model(args, q)
File "E:\code\detection\trainer\train.py", line 221, in run_model
scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, len(train_loader))
TypeError: object of type 'DataLoader' has no len()
[train_2020-04-08-19-38-36] failed