Update README.md

README.md

@@ -1,3 +1,84 @@
 ---
 license: apache-2.0
 ---
+
+
+### epoch=4-step=12512.ckpt 
+* trained for 4 epochs on imagenet
+* final validation accuracy: .64
+* final training accuracy: 0.61
+* git hash: `83330c9ba5ee7b593d8fd6d47dd43c3e4a6051c1`
+* config:
+```
+# lightning.pytorch==2.0.0
+seed_everything: true
+trainer:
+  accelerator: auto
+  strategy: auto
+  devices: auto
+  num_nodes: 1
+  precision: 16-mixed
+  logger: null
+  callbacks: null
+  fast_dev_run: false
+  max_epochs: 5
+  min_epochs: null
+  max_steps: -1
+  min_steps: null
+  max_time: null
+  limit_train_batches: null
+  limit_val_batches: null
+  limit_test_batches: null
+  limit_predict_batches: null
+  overfit_batches: 0.0
+  val_check_interval: 0.1
+  check_val_every_n_epoch: 1
+  num_sanity_val_steps: null
+  log_every_n_steps: 5
+  enable_checkpointing: true
+  enable_progress_bar: null
+  enable_model_summary: null
+  accumulate_grad_batches: 1
+  gradient_clip_val: 1
+  gradient_clip_algorithm: null
+  deterministic: null
+  benchmark: null
+  inference_mode: true
+  use_distributed_sampler: true
+  profiler: null
+  detect_anomaly: false
+  barebones: false
+  plugins: null
+  sync_batchnorm: false
+  reload_dataloaders_every_n_epochs: 0
+  default_root_dir: ckpt/test_insert_at_4
+model:
+  resnet_type: 34
+  is_rq: true
+  quantizer_args:
+    num_quantizers: 4
+    shared_codebook: true
+    quantize_dropout: false
+    accept_image_fmap: true
+    codebook_dim: 128
+    codebook_size: 256
+    decay: 0.8
+    eps: 1.0e-05
+    commitment_weight: 5.0
+    threshold_ema_dead_code: 1
+    sample_codebook_temp: 0.0
+  resnet_insertion_index: 4
+  lr: 0.0002
+  unfreeze_fc: true
+  unfreeze_resnet_block_indeces:
+  - 3
+data:
+  data_dir: /home/figes/Downloads/ILSVRC2012_CLS-LOC/
+  image_size: 224
+  num_workers: 8
+  batch_size: 512
+  shuffle: true
+  pin_memory: true
+  drop_last: false
+ckpt_path: null
+```