amaiasalvador commited on
Commit
9f2a462
·
1 Parent(s): bd07fad

fix: removed perminv_ingrs deprecated argument. Improved instructions.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -90,7 +90,7 @@ We train our model in two stages:
90
 
91
  ```bash
92
  python train.py --model_name im2ingr --batch_size 150 --finetune_after 0 --ingrs_only \
93
- --es_metric iou_sample --perminv_ingrs --loss_weight 0 1000.0 1.0 1.0 \
94
  --learning_rate 1e-4 --scale_learning_rate_cnn 1.0 \
95
  --save_dir ../checkpoints --recipe1m_dir path_to_dataset
96
  ```
@@ -111,7 +111,7 @@ $ tensorboard --logdir='../tb_logs' --port=6006
111
  ### Evaluation
112
 
113
  - Save generated recipes to disk with
114
- ```python sample.py --model_name model --greedy --eval_split test```.
115
  - This script will return ingredient metrics (F1 and IoU)
116
 
117
  ### License
 
90
 
91
  ```bash
92
  python train.py --model_name im2ingr --batch_size 150 --finetune_after 0 --ingrs_only \
93
+ --es_metric iou_sample --loss_weight 0 1000.0 1.0 1.0 \
94
  --learning_rate 1e-4 --scale_learning_rate_cnn 1.0 \
95
  --save_dir ../checkpoints --recipe1m_dir path_to_dataset
96
  ```
 
111
  ### Evaluation
112
 
113
  - Save generated recipes to disk with
114
+ ```python sample.py --model_name model --save_dir ../checkpoints --recipe1m_dir path_to_dataset --greedy --eval_split test```.
115
  - This script will return ingredient metrics (F1 and IoU)
116
 
117
  ### License