config.json

{
  "num_configs": 279,
  "image_embedding_size": 512,
  "pod_embedding_size": 512,
  "dict_size": 119547,
  "config_embedding_size": 16,
  "config_embedding_dropout": 0.2,
  "pod_embedding_dropout": 0.0,
  "dropout1": 0.7,
  "dropout2": 0.35,
  "dropout3": 0.35,
  "fc1_size": 1024,
  "fc2_size": 512,
  "fc3_size": 512,
  "aggregation_mode": "mean",
  "lr": 0.0001,
  "batch_size": 8,
  "weight_decay": 0.5,
  "num_epochs": 15,
  "device": "cpu",
  "generate_embeddings": 0,
  "dataset_load": 1,
  "dataset_size": 100000,
  "lr_milestones": [
    2,
    100
  ],
  "even_by_region": false,
  "with_resources": false,
  "dummy_model": false,
  "find_min": true
}