will-jl944 ed65b01157 convert tensor to list while saving pruner_inputs 4 years ago
..
slim d4f745add9 add dygrah 4 years ago
utils 9ec76631c0 add confusion matrix computation to seg 4 years ago
__init__.py d4f745add9 add dygrah 4 years ago
base.py ed65b01157 convert tensor to list while saving pruner_inputs 4 years ago
classifier.py 9af7b0fe3a all_gather prediction results in classification evaluation 4 years ago
detector.py 2275e8ea59 refine multi-card evaluation logistics, elaborate return value of evaluate only on card 0 4 years ago
load_model.py ed65b01157 convert tensor to list while saving pruner_inputs 4 years ago
segmenter.py 71ed1d640e refine confusion matrix computation 4 years ago