adams-story commited on
Commit
44ab3d5
·
1 Parent(s): d3aa146

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +82 -0
README.md CHANGED
@@ -1,6 +1,88 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  epoch2-val-63.ckpt
5
  * final val acc .63
6
  * trained for 2 epochs
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ epoch=1-step=4503.ckpt
5
+ * inserted at 3, all resnet weights frozen
6
+ * ~.62 val acc
7
+ ```
8
+ # lightning.pytorch==2.0.0
9
+ seed_everything: true
10
+ trainer:
11
+ callbacks:
12
+ - class_path: lightning.pytorch.callbacks.ModelCheckpoint
13
+ init_args:
14
+ save_last: true
15
+ save_top_k: 1
16
+ monitor: v_c_loss
17
+ accelerator: auto
18
+ strategy: auto
19
+ devices: auto
20
+ num_nodes: 1
21
+ precision: 16-mixed
22
+ logger: null
23
+ callbacks: null
24
+ fast_dev_run: false
25
+ max_epochs: 10
26
+ min_epochs: null
27
+ max_steps: -1
28
+ min_steps: null
29
+ max_time: null
30
+ limit_train_batches: null
31
+ limit_val_batches: null
32
+ limit_test_batches: null
33
+ limit_predict_batches: null
34
+ overfit_batches: 0.0
35
+ val_check_interval: 0.1
36
+ check_val_every_n_epoch: 1
37
+ num_sanity_val_steps: null
38
+ log_every_n_steps: 5
39
+ enable_checkpointing: true
40
+ enable_progress_bar: null
41
+ enable_model_summary: null
42
+ accumulate_grad_batches: 1
43
+ gradient_clip_val: 0.5
44
+ gradient_clip_algorithm: null
45
+ deterministic: null
46
+ benchmark: null
47
+ inference_mode: true
48
+ use_distributed_sampler: true
49
+ profiler: null
50
+ detect_anomaly: false
51
+ barebones: false
52
+ plugins: null
53
+ sync_batchnorm: false
54
+ reload_dataloaders_every_n_epochs: 0
55
+ default_root_dir: ckpt/test_insert_at_3_frozen
56
+ model:
57
+ resnet_type: 34
58
+ is_rq: true
59
+ quantizer_args:
60
+ num_quantizers: 4
61
+ shared_codebook: false
62
+ quantize_dropout: true
63
+ accept_image_fmap: true
64
+ codebook_dim: 128
65
+ codebook_size: 512
66
+ decay: 0.85
67
+ eps: 1.0e-05
68
+ commitment_weight: 25.0
69
+ threshold_ema_dead_code: 2
70
+ sample_codebook_temp: 0.05
71
+ quantize_dropout_cutoff_index: 1
72
+ quantize_dropout_multiple_of: 1
73
+ resnet_insertion_index: 3
74
+ lr: 0.0002
75
+ data:
76
+ data_dir: "/home/figes/Downloads/ILSVRC2012_CLS-LOC/"
77
+ image_size: 224
78
+ num_workers: 8
79
+ batch_size: 512
80
+ shuffle: true
81
+ pin_memory: true
82
+ drop_last: false
83
+ ```
84
+
85
+
86
  epoch2-val-63.ckpt
87
  * final val acc .63
88
  * trained for 2 epochs