Upload folder using huggingface_hub
Browse files- config.yaml +2 -2
- model.safetensors +1 -1
config.yaml
CHANGED
|
@@ -2,12 +2,12 @@ resume: false
|
|
| 2 |
device: cuda
|
| 3 |
use_amp: false
|
| 4 |
seed: 1000
|
| 5 |
-
dataset_repo_id: pranavsaroha/
|
| 6 |
video_backend: pyav
|
| 7 |
training:
|
| 8 |
offline_steps: 80000
|
| 9 |
num_workers: 4
|
| 10 |
-
batch_size:
|
| 11 |
eval_freq: -1
|
| 12 |
log_freq: 100
|
| 13 |
save_checkpoint: true
|
|
|
|
| 2 |
device: cuda
|
| 3 |
use_amp: false
|
| 4 |
seed: 1000
|
| 5 |
+
dataset_repo_id: pranavsaroha/so100_squishy2colors_1
|
| 6 |
video_backend: pyav
|
| 7 |
training:
|
| 8 |
offline_steps: 80000
|
| 9 |
num_workers: 4
|
| 10 |
+
batch_size: 4
|
| 11 |
eval_freq: -1
|
| 12 |
log_freq: 100
|
| 13 |
save_checkpoint: true
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 206701072
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:909cb595660bd087bde34f24f6c9adc852c082a1a58f33a70f9786932516a052
|
| 3 |
size 206701072
|