chenz53 commited on
Commit
7548a0b
·
verified ·
1 Parent(s): 99aa071

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +73 -3
README.md CHANGED
@@ -1,3 +1,73 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ ---
6
+ library_name: transformers
7
+ tags:
8
+ - masked-image-modeling
9
+ - generated_from_trainer
10
+ model-index:
11
+ - name: smb-vision-base-1029
12
+ results: [0.5023 L2-loss]
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # smb-vision-base-1029
19
+
20
+ This model is trained from scratch using [VideoMAE](https://huggingface.co/docs/transformers/en/model_doc/videomae) on over 4.7k CT volumes.
21
+
22
+ ## Model description
23
+
24
+ More information needed
25
+
26
+ ## Intended uses & limitations
27
+
28
+ More information needed
29
+
30
+ ## Training and evaluation data
31
+
32
+ More information needed
33
+
34
+ ## Training procedure
35
+
36
+ ### Training hyperparameters
37
+
38
+ The following hyperparameters were used during training:
39
+ - learning_rate: 3e-04
40
+ - train_batch_size: 32
41
+ - eval_batch_size: 1
42
+ - seed: 42
43
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
44
+ - lr_scheduler_type: cosine
45
+ - num_epochs: 30.0
46
+
47
+ ### Training results
48
+
49
+ {
50
+ "_runtime": 54805.860011105,
51
+ "_step": 4351,
52
+ "eval/runtime": 17.8428,
53
+ "eval/samples_per_second": 2.578,
54
+ "eval/steps_per_second": 2.578,
55
+ "total_flos": 3.8084565648770335e+21,
56
+ "train/epoch": 30,
57
+ "train/global_step": 4350,
58
+ "train/grad_norm": 0.0735374316573143,
59
+ "train/learning_rate": 0,
60
+ "train/loss": 0.5736,
61
+ "train_loss": 0.5022664608695041,
62
+ "train_runtime": 54785.1298,
63
+ "train_samples_per_second": 2.527,
64
+ "train_steps_per_second": 0.079
65
+ }
66
+
67
+
68
+ ### Framework versions
69
+
70
+ - Transformers 4.46.0
71
+ - Pytorch 2.5.0
72
+ - Datasets 3.0.2
73
+ - Tokenizers 0.20.1