Skip to content

Commit 1770e3e

Browse files
authored
not using advprop by default
1 parent a1001a8 commit 1770e3e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnet/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def from_name(cls, model_name, override_params=None):
213213
return cls(blocks_args, global_params)
214214

215215
@classmethod
216-
def from_pretrained(cls, model_name, load_weights=True, advprop=True, num_classes=1000, in_channels=3):
216+
def from_pretrained(cls, model_name, load_weights=True, advprop=False, num_classes=1000, in_channels=3):
217217
model = cls.from_name(model_name, override_params={'num_classes': num_classes})
218218
if load_weights:
219219
load_pretrained_weights(model, model_name, load_fc=(num_classes == 1000), advprop=advprop)

0 commit comments

Comments
 (0)