amaiasalvador commited on
Commit
9286344
·
1 Parent(s): 3ab629c

ingredient encoder parameters removed when args.ingrs_only = True

Browse files
Files changed (1) hide show
  1. 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()) + list(model.ingredient_encoder.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: