Commit Graph
9 Commits
Author SHA1 Message Date
geohot 892ac661e1 enet readability 2022-06-07 10:23:05 -07:00
geohot e28cdfb0cf clean up resnet 2021-11-30 16:14:54 -05:00
geohot 8f5779eeaa very minor change 2021-11-30 15:54:03 -05:00
geohot 46bbbcf7f0 model touchups 2021-11-30 11:13:34 -05:00
geohot 908db3bdea support bias in conv like linear 2021-11-30 00:44:59 -05:00
geohot bd21304e3c linear takes in weight and bias 2021-11-30 00:38:47 -05:00
geohot 7d12482d80 refactor efficientnet loading 2021-10-30 17:02:17 -07:00
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 <[email protected]>
2021-10-22 15:10:04 -07:00
geohot d3f169b267 move good models to models, add a training step test 2021-06-19 11:24:15 -07:00