Skip to content

Commit

Permalink
Fix: "cut" parameter not passed to vision_learner
Browse files Browse the repository at this point in the history
  • Loading branch information
adamamer20 committed Apr 25, 2024
1 parent 37bc4db commit e9736c3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastai/vision/learner.py
Expand Up @@ -225,7 +225,7 @@ def vision_learner(dls, arch, normalize=True, n_out=None, pretrained=True, weigh
if n_out is None: n_out = get_c(dls)
assert n_out, "`n_out` is not defined, and could not be inferred from data, set `dls.c` or pass `n_out`"
meta = model_meta.get(arch, _default_meta)
model_args = dict(init=init, custom_head=custom_head, concat_pool=concat_pool, pool=pool, lin_ftrs=lin_ftrs, ps=ps,
model_args = dict(cut=cut, init=init, custom_head=custom_head, concat_pool=concat_pool, pool=pool, lin_ftrs=lin_ftrs, ps=ps,
first_bn=first_bn, bn_final=bn_final, lin_first=lin_first, y_range=y_range, **kwargs)
n_in = kwargs['n_in'] if 'n_in' in kwargs else 3
if isinstance(arch, str):
Expand Down

0 comments on commit e9736c3

Please sign in to comment.