amaiasalvador
commited on
Commit
·
9f2a462
1
Parent(s):
bd07fad
fix: removed perminv_ingrs deprecated argument. Improved instructions.
Browse files
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 --
|
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
|