Hugging Face's logo

Spaces:
Nada0salim
/
image-captioning
No application file

App Files Community
image-captioning
File size: 725 Bytes
becb4f7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
{
  "data": {
    "images_dir": "data/raw/Images",
    "captions_file": "data/raw/captions.txt",
    "features_dir": "data/processed",
    "vocab_path": "data/processed/vocab.json",
    "splits_path": "data/processed/splits.json",
    "min_word_freq": 3,
    "max_caption_len": 35
  },
  "model": {
    "encoder": "resnet50",
    "embed_dim": 256,
    "hidden_dim": 512,
    "num_layers": 1,
    "dropout": 0.5
  },
  "training": {
    "batch_size": 64,
    "num_epochs": 30,
    "learning_rate": 0.001,
    "weight_decay": 0.0001,
    "patience": 5,
    "teacher_forcing": 1.0,
    "grad_clip": 5.0,
    "device": "cuda",
    "num_workers": 2
  },
  "checkpoint": {
    "save_dir": "checkpoints",
    "save_best": true
  }
}