Skip to content

Commit

Permalink
Fix passing arg to constructor
Browse files Browse the repository at this point in the history
  • Loading branch information
radekosmulski committed Feb 16, 2018
1 parent 7b1345a commit a252c55
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastai/conv_learner.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def __init__(self, data, models, precompute=False, **kwargs):
@classmethod
def pretrained(cls, f, data, ps=None, xtra_fc=None, xtra_cut=0, precompute=False, **kwargs):
models = ConvnetBuilder(f, data.c, data.is_multi, data.is_reg, ps=ps, xtra_fc=xtra_fc, xtra_cut=xtra_cut)
return cls(data, models, **kwargs)
return cls(data, models, precompute, **kwargs)

@property
def model(self): return self.models.fc_model if self.precompute else self.models.model
Expand Down

0 comments on commit a252c55

Please sign in to comment.