JensLundsgaard commited on
Commit
d87865d
·
verified ·
1 Parent(s): 9b67c2b

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +30 -9
config.json CHANGED
@@ -1,16 +1,37 @@
1
  {
2
- "decoder_hidden_dim": 128,
3
- "decoder_layers": 2,
4
- "dropout_rate": 0.1,
 
5
  "encoder_hidden_dim": 256,
6
  "encoder_layers": 2,
7
- "input_channels": 1,
 
8
  "latent_size": 4096,
9
- "num_classes": 2,
10
- "seq_len": 50,
11
- "use_batchnorm": true,
12
  "use_classifier": false,
13
- "use_convlstm": true,
14
  "use_latent_split": false,
15
- "use_residual": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  }
 
1
  {
2
+ "model_type": "ConvLSTMAutoencoder",
3
+ "architecture": "ConvLSTM Autoencoder",
4
+ "seq_len": 50,
5
+ "input_channels": 1,
6
  "encoder_hidden_dim": 256,
7
  "encoder_layers": 2,
8
+ "decoder_hidden_dim": 128,
9
+ "decoder_layers": 2,
10
  "latent_size": 4096,
 
 
 
11
  "use_classifier": false,
12
+ "num_classes": 2,
13
  "use_latent_split": false,
14
+ "image_size": 128,
15
+ "dropout_rate": 0.1,
16
+ "use_convlstm": true,
17
+ "use_residual": true,
18
+ "use_batchnorm": true,
19
+ "loss_type": "l1",
20
+ "ms_ssim_weight": 1.0,
21
+ "rec_weight": 0.0,
22
+ "temporal_weight": 0.05,
23
+ "loss_description": "MS-SSIM(1.0) + Temporal(0.05)",
24
+ "learning_rate": 0.0002,
25
+ "weight_decay": 1e-05,
26
+ "optimizer": "Adam",
27
+ "scheduler": "CosineAnnealingLR",
28
+ "batch_size": 1,
29
+ "epochs": 10,
30
+ "gradient_clip": 5.0,
31
+ "dataset": "https://zenodo.org/records/7912264",
32
+ "resize": 128,
33
+ "normalization": "minmax01",
34
+ "repo_name": "embryo-convlstm-c477cc64-2026-02-18",
35
+ "date": "2026-02-18",
36
+ "hash": "02"
37
  }