Name | Last commit | Last update |
---|---|---|
.. | ||
convert_datasets | ||
benchmark.py | ||
dist_test.sh | ||
dist_train.sh | ||
get_flops.py | ||
print_config.py | ||
publish_model.py | ||
pytorch2onnx.py | ||
slurm_test.sh | ||
slurm_train.sh | ||
test.py | ||
train.py |
* memory efficient test * implement efficient test * merge * Add document and docstring * fix unit test * add memory usage report
Name | Last commit | Last update |
---|---|---|
.. | ||
convert_datasets | ||
benchmark.py | ||
dist_test.sh | ||
dist_train.sh | ||
get_flops.py | ||
print_config.py | ||
publish_model.py | ||
pytorch2onnx.py | ||
slurm_test.sh | ||
slurm_train.sh | ||
test.py | ||
train.py |