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

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. config.json +9 -30
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,37 +1,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
  }
 
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
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:49e3ecd450142752622c913b252f70d2c89f6bcc7009f146fbbba3e8396b8be3
3
  size 2234344348
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90639b59defc808158ff4931a67a3ec95359e317c02ac1908e57d4804e009d93
3
  size 2234344348