Sebastian Kreft 3358770182 chore(efficientnet): don't use eval when loading weights (#286) ...

Because the weights are being loaded from a third party internet address, it's unsafe to use eval. Also with the change I think the code became a little bit more clear as now it's clearer which keys are being transformed.

Co-authored-by: Seba Kreft <sebastian.kreft@houm.com>

2021-10-22 15:10:04 -07:00

..

efficientnet.py

chore(efficientnet): don't use eval when loading weights (#286)

2021-10-22 15:10:04 -07:00

resnet.py

Added ResNet-{18, 34, 50, 101, 152} (#271)

2021-06-21 09:37:24 -07:00

transformer.py

move good models to models, add a training step test

2021-06-19 11:24:15 -07:00