* WIP: vis * WIP: add visualization * WIP: add visualization hook * WIP: support razor visualizer * WIP * WIP: wrap draw_featmap * support feature map visualization * add a demo image for visualization * fix typos * change eps to 1e-6 * add pytest for visualization * fix vis hook * fix arguments' name * fix img path * support draw inference results * add visualization doc * fix figure url * move files Co-authored-by: weihan cao <HIT-cwh> |
||
---|---|---|
.. | ||
misc | ||
model_converters | ||
visualizations | ||
dist_test.sh | ||
dist_train.sh | ||
get_channel_units.py | ||
slurm_test.sh | ||
slurm_train.sh | ||
test.py | ||
train.py |