amaiasalvador
commited on
Commit
·
9286344
1
Parent(s):
3ab629c
ingredient encoder parameters removed when args.ingrs_only = True
Browse files- src/train.py +1 -1
src/train.py
CHANGED
@@ -144,7 +144,7 @@ def main(args):
|
|
144 |
|
145 |
# add model parameters
|
146 |
if args.ingrs_only:
|
147 |
-
params = list(model.ingredient_decoder.parameters())
|
148 |
elif args.recipe_only:
|
149 |
params = list(model.recipe_decoder.parameters()) + list(model.ingredient_encoder.parameters())
|
150 |
else:
|
|
|
144 |
|
145 |
# add model parameters
|
146 |
if args.ingrs_only:
|
147 |
+
params = list(model.ingredient_decoder.parameters())
|
148 |
elif args.recipe_only:
|
149 |
params = list(model.recipe_decoder.parameters()) + list(model.ingredient_encoder.parameters())
|
150 |
else:
|