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:
|