Commit Graph

30 Commits (01538bec6402b6e4b15f8cb07e1b9c2a3d2ef7a7)

Author SHA1 Message Date
zuchen.wang 01538bec64 fix logger and pretty dataset info 2021-11-04 10:11:16 +08:00
zuchen.wang 21c14f1494 重构数据集 2021-11-03 17:25:58 +08:00
zuchen.wang 65cfc515d9 实现和线上模型一模一样的PCBNet 2021-11-02 17:56:05 +08:00
zuchen.wang 5a075c1fe8 Merge branch 'develop' into pcb_online 2021-10-29 14:29:29 +08:00
zuchen.wang d03eae9daa Implement online pcb config 2021-10-29 14:13:45 +08:00
zuchen.wang c90792042a make common base config: input, dataloader, solver, test period and batch 2021-10-29 11:45:24 +08:00
zuchen.wang eabe896ee8 train pcb, 未达到期望 2021-10-28 22:39:54 +08:00
zuchen.wang ad31a2cd16 modify config 2021-10-27 16:15:58 +08:00
zuchen.wang 3880b23ab4 refactor evaluator 2021-10-27 16:03:43 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
zuchen.wang 31d7a51b62 fix bug in online_dataset.py 2021-10-22 13:31:12 +08:00
zuchen.wang ae842114c7 update dataset config in trainer.py 2021-10-21 10:50:38 +08:00
zuchen.wang 0dd493b7b4 make pair dataset negative sample from full set 2021-10-20 17:04:27 +08:00
zuchen.wang 6ab4ac9784 modify logger 2021-10-18 17:38:08 +08:00
zuchen.wang e4c46cbfed add r101-se-ibn.yaml 2021-10-18 17:37:41 +08:00
zuchen.wang 3788c5484b add excel data set 2021-10-18 15:37:25 +08:00
zuchen.wang 3e753612a9 change pair evaluator 2021-10-18 15:36:54 +08:00
zuchen.wang 7190348568 fix pair_dataset random seed in val and test mode
add ce loss
2021-10-16 21:42:32 +08:00
zuchen.wang 1feda07ce6 evaluation success 2021-10-16 21:24:16 +08:00
zuchen.wang 786c442391 add bughook in train_net.py 2021-10-16 20:56:01 +08:00
zuchen.wang d87f4a8419 add evaluator in fastshoe trainner 2021-10-16 00:03:00 +08:00
zuchen.wang d162c5fc4e add PairHead 2021-10-15 21:26:16 +08:00
zuchen.wang 8c09ec0d3d update forward 2021-10-13 20:05:08 +08:00
zuchen.wang d2ee44d503 change cls_target to clas_target 2021-10-13 20:05:08 +08:00
zuchen.wang 1bd41a79fc add cls target in pair dataset 2021-10-13 20:05:07 +08:00
zuchen.wang 8b309b0f4e add contrastive loss 2021-10-13 20:05:07 +08:00
zuchen.wang 7201a82840 add pair collator_fn 2021-10-13 20:05:07 +08:00
zuchen.wang bced0d04ff override some method in dataset 2021-10-13 20:05:07 +08:00
zuchen.wang e9cd5311f5 add some paths 2021-10-13 20:05:07 +08:00
zuchen.wang 46c1ce83a7 add fastshoe 2021-10-13 20:05:07 +08:00