hwangdeyu 7a4be5c035 add operator HardSwishJitAutoFn export to onnx

2021-02-03 09:06:53 +08:00

..

data

Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py

2021-01-22 16:54:33 -08:00

loss

…

models

add operator HardSwishJitAutoFn export to onnx

2021-02-03 09:06:53 +08:00

optim

Configure create_optimizer with args.opt_args

2020-12-08 00:03:09 -06:00

scheduler

…

utils

Fix #387 so that checkpoint saver works with max history of 1. Add checkpoint-hist arg to train.py.

2021-01-31 20:14:51 -08:00

__init__.py

…

version.py

Update README.md for NF-nets, bump version to 0.4.1 for merge

2021-01-30 23:19:45 -08:00