FOTS.PyTorch icon indicating copy to clipboard operation
FOTS.PyTorch copied to clipboard

Recognition Loss: 0.000000 ?

Open 1145520074 opened this issue 3 years ago • 1 comments

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

1145520074 avatar Apr 09 '21 11:04 1145520074

maybe because your mode is 'detection'. set it to 'united'

xansar avatar Jun 07 '21 09:06 xansar