TF-Keras
category_classifier / config.json
raphael0202's picture
Upload folder using huggingface_hub
50cf680 verified
{
"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
}