FOTS.PyTorch
FOTS.PyTorch copied to clipboard
Recognition Loss: 0.000000 ?
Train Epoch: 1 [800/928 (86%)] Loss: 0.022387 IOU Loss: 0.012452 CLS Loss: 0.009935 Recognition Loss: 0.000000 epoch : 1 loss : 0.029173220796831722 precious : 0.0 recall : 0.0 hmean : 0.0 val_precious : 0.0 val_recall : 0.0 val_hmean : 0.0 Saving current best: model_best.pth.tar ...
It is strange? my config is : { "name": "FOTS", "cuda": true, "gpus": [0], "data_loader": { "dataset":"icdar2015", "data_dir": "/data/zhaoxiang/Data/icdar2015/", "batch_size": 32, "shuffle": true, "workers": 4 }, "validation": { "validation_split": 0.1, "shuffle": true },
"lr_scheduler_type": "ExponentialLR",
"lr_scheduler_freq": 10000,
"lr_scheduler": {
"gamma": 0.94
},
"optimizer_type": "Adam",
"optimizer": {
"lr": 0.0001,
"weight_decay": 1e-5
},
"loss": "FOTSLoss",
"metrics": ["fots_metric"],
"trainer": {
"epochs": 100000,
"save_dir": "saved/",
"save_freq": 10,
"verbosity": 2,
"monitor": "hmean",
"monitor_mode": "max"
},
"arch": "FOTSModel",
"model": {
"mode": "detection"
}
}
Thanks very much
maybe because your mode is 'detection'. set it to 'united'