[2022/03/06 03:27:03] root INFO: epoch: [35/5000], iter: 5990, lr: 0.000100, loss: 1.938344, loss_shrink_maps: 1.091385, loss_threshold_maps: 0.636650, loss_binary_maps: 0.212405, reader_cost: 0.40183 s, batch_cost: 0.65372 s, samples: 80, ips: 12.23763
[2022/03/06 03:27:11] root INFO: epoch: [35/5000], iter: 6000, lr: 0.000100, loss: 2.023747, loss_shrink_maps: 1.144246, loss_threshold_maps: 0.646133, loss_binary_maps: 0.223270, reader_cost: 0.56848 s, batch_cost: 0.81862 s, samples: 80, ips: 9.77254
eval model:: 0%| | 0/5097 [00:00eval model:: 0%| | 1/5097 [00:04<6:19:56, 4.47s/it]
eval model:: 0%| | 2/5097 [00:05<4:56:24, 3.49s/it]
用的是v100至尊版
MobileNetV3+DBNet
验证配置
loader:
shuffle: False
drop_last: False
batch_size_per_card: 1 # must be 1
num_workers: 8
use_shared_memory: False
评估部分确实耗时较多,如果觉得不需要,也可以注释掉eval()