alsalivan commited on
Commit
d032e73
·
1 Parent(s): 9d4d50c

Initial model upload with safetensors and config

Browse files
Files changed (2) hide show
  1. config.json +39 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/vol/aimspace/users/seliv/Documents/projects/output/ecgcmr/experiments/2024-11-05/12-15-52/best_val_loss_hf_model",
3
+ "architectures": [
4
+ "ViTMAEModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "decoder_hidden_size": 256,
8
+ "decoder_intermediate_size": 1024,
9
+ "decoder_num_attention_heads": 4,
10
+ "decoder_num_hidden_layers": 1,
11
+ "drop_path_rate": 0.1,
12
+ "hidden_act": "gelu",
13
+ "hidden_dropout_prob": 0.1,
14
+ "hidden_size": 512,
15
+ "image_size": [
16
+ 12,
17
+ 2500
18
+ ],
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 2048,
21
+ "layer_norm_eps": 1e-12,
22
+ "layerscale_init_values": 1e-05,
23
+ "mask_loss": true,
24
+ "mask_ratio": 0.8,
25
+ "model_type": "vit_mae",
26
+ "norm_pix_loss": false,
27
+ "num_attention_heads": 8,
28
+ "num_channels": 1,
29
+ "num_hidden_layers": 6,
30
+ "patch_size": [
31
+ 1,
32
+ 50
33
+ ],
34
+ "qkv_bias": true,
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.41.0.dev0",
37
+ "use_cls_token": true,
38
+ "use_learnable_pos_emb": true
39
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c118529422441651bd02e153325ad2088500ba80938dac09e732a83a017dd35
3
+ size 77022800