Commit
•
9d33806
1
Parent(s):
5d6ce37
Update README.md
Browse files
README.md
CHANGED
|
@@ -3,16 +3,22 @@ license: apache-2.0
|
|
| 3 |
---
|
| 4 |
|
| 5 |
|
| 6 |
-
### epoch=
|
| 7 |
-
* trained for
|
| 8 |
-
* final validation accuracy: .
|
| 9 |
-
* final training accuracy: 0.
|
| 10 |
-
* git hash: `
|
| 11 |
* config:
|
| 12 |
```
|
| 13 |
# lightning.pytorch==2.0.0
|
| 14 |
seed_everything: true
|
| 15 |
trainer:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
accelerator: auto
|
| 17 |
strategy: auto
|
| 18 |
devices: auto
|
|
@@ -21,7 +27,7 @@ trainer:
|
|
| 21 |
logger: null
|
| 22 |
callbacks: null
|
| 23 |
fast_dev_run: false
|
| 24 |
-
max_epochs:
|
| 25 |
min_epochs: null
|
| 26 |
max_steps: -1
|
| 27 |
min_steps: null
|
|
@@ -39,7 +45,7 @@ trainer:
|
|
| 39 |
enable_progress_bar: null
|
| 40 |
enable_model_summary: null
|
| 41 |
accumulate_grad_batches: 1
|
| 42 |
-
gradient_clip_val:
|
| 43 |
gradient_clip_algorithm: null
|
| 44 |
deterministic: null
|
| 45 |
benchmark: null
|
|
@@ -68,17 +74,16 @@ model:
|
|
| 68 |
threshold_ema_dead_code: 1
|
| 69 |
sample_codebook_temp: 0.0
|
| 70 |
resnet_insertion_index: 4
|
| 71 |
-
lr: 0.0002
|
| 72 |
-
unfreeze_fc: true
|
| 73 |
unfreeze_resnet_block_indeces:
|
| 74 |
-
|
|
|
|
|
|
|
| 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 |
-
ckpt_path: null
|
| 84 |
```
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
|
| 6 |
+
### epoch=5-step=14765.ckpt
|
| 7 |
+
* trained for 5 1/2 epochs on imagenet, on top of resnet 34
|
| 8 |
+
* final validation accuracy: .66
|
| 9 |
+
* final training accuracy: 0.64
|
| 10 |
+
* git hash: `c4852331f9a40393b8ffd8b7b9a689d1ff6e1021`
|
| 11 |
* config:
|
| 12 |
```
|
| 13 |
# lightning.pytorch==2.0.0
|
| 14 |
seed_everything: true
|
| 15 |
trainer:
|
| 16 |
+
callbacks:
|
| 17 |
+
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
|
| 18 |
+
init_args:
|
| 19 |
+
save_last: true
|
| 20 |
+
save_top_k: 1
|
| 21 |
+
monitor: v_c_loss
|
| 22 |
accelerator: auto
|
| 23 |
strategy: auto
|
| 24 |
devices: auto
|
|
|
|
| 27 |
logger: null
|
| 28 |
callbacks: null
|
| 29 |
fast_dev_run: false
|
| 30 |
+
max_epochs: 10
|
| 31 |
min_epochs: null
|
| 32 |
max_steps: -1
|
| 33 |
min_steps: null
|
|
|
|
| 45 |
enable_progress_bar: null
|
| 46 |
enable_model_summary: null
|
| 47 |
accumulate_grad_batches: 1
|
| 48 |
+
gradient_clip_val: 0.5
|
| 49 |
gradient_clip_algorithm: null
|
| 50 |
deterministic: null
|
| 51 |
benchmark: null
|
|
|
|
| 74 |
threshold_ema_dead_code: 1
|
| 75 |
sample_codebook_temp: 0.0
|
| 76 |
resnet_insertion_index: 4
|
|
|
|
|
|
|
| 77 |
unfreeze_resnet_block_indeces:
|
| 78 |
+
- 3
|
| 79 |
+
unfreeze_fc: true
|
| 80 |
+
lr: 0.0002
|
| 81 |
data:
|
| 82 |
+
data_dir: "/home/figes/Downloads/ILSVRC2012_CLS-LOC/"
|
| 83 |
image_size: 224
|
| 84 |
num_workers: 8
|
| 85 |
batch_size: 512
|
| 86 |
shuffle: true
|
| 87 |
pin_memory: true
|
| 88 |
drop_last: false
|
|
|
|
| 89 |
```
|