7 Commits (a5dac37beafc39fb47fe947ee3595eb0fe794794)

Author	SHA1	Message	Date
liaoxingyu	8d4bd77ae3	fix visualization problem #233	2020-08-20 17:03:43 +08:00
liaoxingyu	16655448c2	onnx/trt support ... Summary: change model pretrain mode and support onnx/TensorRT export	2020-07-29 17:43:39 +08:00
liaoxingyu	fec7abc461	finish v0.2 ddp training	2020-07-06 16:57:43 +08:00
liaoxingyu	9bef814155	update visulize logger	2020-05-27 22:53:27 +08:00
liaoxingyu	84c733fa85	fix: remove prefetcher, put normalizer in model ... 1. remove messy data prefetcher which will cause confusion 2. put normliazer in model to accelerate training via GPU computing	2020-05-25 23:39:11 +08:00
liaoxingyu	2ac55a7601	feat: update roc curve and TPR@FPR metric ... support plot multiple ROC curves with different model	2020-05-20 14:29:33 +08:00
liaoxingyu	e344eae1cc	feat: support plotting roc curve and compute auc score ... ROC curve and AUC score will be help for thresholds	2020-05-19 20:45:26 +08:00