tinygrad/models
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