@@ -276,12 +276,14 @@ def get_model_params(model_name, override_params):
276276
277277
278278url_map = {
279- 'efficientnet-b0' : 'http://storage.googleapis.com/public-models/efficientnet-b0-08094119.pth' ,
280- 'efficientnet-b1' : 'http://storage.googleapis.com/public-models/efficientnet-b1-dbc7070a.pth' ,
281- 'efficientnet-b2' : 'http://storage.googleapis.com/public-models/efficientnet-b2-27687264.pth' ,
282- 'efficientnet-b3' : 'http://storage.googleapis.com/public-models/efficientnet-b3-c8376fa2.pth' ,
283- 'efficientnet-b4' : 'http://storage.googleapis.com/public-models/efficientnet-b4-e116e8b3.pth' ,
284- 'efficientnet-b5' : 'http://storage.googleapis.com/public-models/efficientnet-b5-586e6cc6.pth' ,
279+ 'efficientnet-b0' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b0-355c32eb.pth' ,
280+ 'efficientnet-b1' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b1-f1951068.pth' ,
281+ 'efficientnet-b2' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b2-8bb594d6.pth' ,
282+ 'efficientnet-b3' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b3-5fb5a3c3.pth' ,
283+ 'efficientnet-b4' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b4-6ed6700e.pth' ,
284+ 'efficientnet-b5' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b5-b6417697.pth' ,
285+ 'efficientnet-b6' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b6-c76e70fd.pth' ,
286+ 'efficientnet-b7' : 'http://storage.googleapis.com/public-models/efficientnet/efficientnet-b7-dcc49843.pth' ,
285287}
286288
287289def load_pretrained_weights (model , model_name , load_fc = True ):
0 commit comments