| { | |
| "name": "add-image-embedding-300-epochs-8-heads", | |
| "epochs": 300, | |
| "batch_size": 128, | |
| "learning_rate": 0.001, | |
| "mixed_precision": false, | |
| "random_seed": 42, | |
| "category_min_count": 10, | |
| "add_product_name_input": true, | |
| "add_ingredient_ocr_input": true, | |
| "add_ingredient_input": true, | |
| "add_nutriment_input": true, | |
| "add_image_embedding_input": true, | |
| "nutriment_num_bins": 100, | |
| "ingredient_min_freq": 3, | |
| "ingredient_ocr_min_freq": 3, | |
| "ingredient_embedding_size": 64, | |
| "ingredient_output_size": 64, | |
| "ingredient_recurrent": false, | |
| "ingredient_lstm_dropout": 0.2, | |
| "product_name_max_tokens": 93000, | |
| "product_name_output_sequence_length": 30, | |
| "product_name_split_method": "whitespace", | |
| "product_name_embedding_size": 64, | |
| "product_name_output_size": 64, | |
| "product_name_lstm_dropout": 0.2, | |
| "image_embedding_key_dim": 64, | |
| "image_embedding_num_heads": 8, | |
| "dense_layer_dropout": 0.2, | |
| "dense_layer_output_size": 64, | |
| "label_smoothing": 0.0, | |
| "cosine_scheduler": true | |
| } |