kevinoli commited on
Commit
aaffa45
1 Parent(s): f262f58

Training in progress, step 500, checkpoint

Browse files
checkpoint-500/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai/clip-vit-large-patch14-336",
3
+ "architectures": [
4
+ "CLIPModel"
5
+ ],
6
+ "initializer_factor": 1.0,
7
+ "logit_scale_init_value": 2.6592,
8
+ "model_type": "clip",
9
+ "projection_dim": 768,
10
+ "text_config": {
11
+ "dropout": 0.0,
12
+ "hidden_size": 768,
13
+ "intermediate_size": 3072,
14
+ "model_type": "clip_text_model",
15
+ "num_attention_heads": 12,
16
+ "projection_dim": 768
17
+ },
18
+ "torch_dtype": "float32",
19
+ "transformers_version": "4.45.0.dev0",
20
+ "vision_config": {
21
+ "dropout": 0.0,
22
+ "hidden_size": 1024,
23
+ "image_size": 336,
24
+ "intermediate_size": 4096,
25
+ "model_type": "clip_vision_model",
26
+ "num_attention_heads": 16,
27
+ "num_hidden_layers": 24,
28
+ "patch_size": 14,
29
+ "projection_dim": 768
30
+ }
31
+ }
checkpoint-500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1b46f76e88ec6d2244ddb394c5b13a4a6ba9182c217592f9639b62500ee70cb
3
+ size 1711848436
checkpoint-500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62e19b25cf423a079f4464a07645a4a57741290c7888409e0ce7f3403701224e
3
+ size 3424043887
checkpoint-500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cddbf207acff68da0d051345656b78fe58e07a53e2d0a8f4d9ea5c8255f66ff
3
+ size 14503
checkpoint-500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fddee935ed9db58de0422a9dd0c6b9c27c639523b5fb1ce4d35213cce4cac708
3
+ size 623
checkpoint-500/trainer_state.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.108838677406311,
3
+ "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l17-l/checkpoint-500",
4
+ "epoch": 0.05332764505119454,
5
+ "eval_steps": 500,
6
+ "global_step": 500,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.05332764505119454,
13
+ "grad_norm": 397.1771240234375,
14
+ "learning_rate": 4.933340443686006e-07,
15
+ "loss": 0.3803,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 0.05332764505119454,
20
+ "eval_loss": 1.108838677406311,
21
+ "eval_runtime": 63.8445,
22
+ "eval_samples_per_second": 15.459,
23
+ "eval_steps_per_second": 1.942,
24
+ "step": 500
25
+ }
26
+ ],
27
+ "logging_steps": 500,
28
+ "max_steps": 37504,
29
+ "num_input_tokens_seen": 0,
30
+ "num_train_epochs": 4,
31
+ "save_steps": 500,
32
+ "stateful_callbacks": {
33
+ "TrainerControl": {
34
+ "args": {
35
+ "should_epoch_stop": false,
36
+ "should_evaluate": false,
37
+ "should_log": false,
38
+ "should_save": true,
39
+ "should_training_stop": false
40
+ },
41
+ "attributes": {}
42
+ }
43
+ },
44
+ "total_flos": 180598692332520.0,
45
+ "train_batch_size": 2,
46
+ "trial_name": null,
47
+ "trial_params": null
48
+ }
checkpoint-500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1377635e939bb7bf36d8c4ca52590b78a9c7469dee2f9240633030e35d7f405a
3
+ size 4847