Commit Graph
2 Commits
Author SHA1 Message Date
Francis LataandGitHub 5755ac1f72 Fix FC layer ResNet load_from_pretrained error (#8387)
* validate that FC exists before loading pretrained weights

* add test case for ResNet pretrained model without FC layer

* remove extra newline

* rename test case

* reraise exception if not handled by check
2024-12-26 18:11:27 -05:00
chenyuandGitHub e2c5054bdd update resnet.load_from_pretrained (#5040) 2024-06-18 16:29:22 -04:00