Upload folder using huggingface_hub
Browse files- checkpoints/020000/pretrained_model/config.json +3 -4
- checkpoints/020000/pretrained_model/model.safetensors +2 -2
- checkpoints/020000/pretrained_model/train_config.json +3 -3
- checkpoints/020000/training_state/optimizer_state.safetensors +2 -2
- checkpoints/020000/training_state/rng_state.safetensors +1 -1
- checkpoints/040000/pretrained_model/config.json +3 -4
- checkpoints/040000/pretrained_model/model.safetensors +2 -2
- checkpoints/040000/pretrained_model/train_config.json +3 -3
- checkpoints/040000/training_state/optimizer_state.safetensors +2 -2
- checkpoints/040000/training_state/rng_state.safetensors +1 -1
- checkpoints/060000/pretrained_model/config.json +3 -4
- checkpoints/060000/pretrained_model/model.safetensors +2 -2
- checkpoints/060000/pretrained_model/train_config.json +3 -3
- checkpoints/060000/training_state/optimizer_state.safetensors +2 -2
- checkpoints/060000/training_state/rng_state.safetensors +1 -1
- checkpoints/080000/pretrained_model/config.json +3 -4
- checkpoints/080000/pretrained_model/model.safetensors +2 -2
- checkpoints/080000/pretrained_model/train_config.json +3 -3
- checkpoints/080000/training_state/optimizer_state.safetensors +2 -2
- checkpoints/080000/training_state/rng_state.safetensors +1 -1
- checkpoints/100000/pretrained_model/config.json +3 -4
- checkpoints/100000/pretrained_model/model.safetensors +2 -2
- checkpoints/100000/pretrained_model/train_config.json +3 -3
- checkpoints/100000/training_state/optimizer_state.safetensors +2 -2
- checkpoints/100000/training_state/rng_state.safetensors +1 -1
checkpoints/020000/pretrained_model/config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
@@ -56,8 +55,8 @@
|
|
56 |
},
|
57 |
"device": "cuda",
|
58 |
"use_amp": false,
|
59 |
-
"chunk_size":
|
60 |
-
"n_action_steps":
|
61 |
"vision_backbone": "resnet18",
|
62 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
63 |
"replace_final_stride_with_dilation": false,
|
@@ -71,7 +70,7 @@
|
|
71 |
"use_vae": true,
|
72 |
"latent_dim": 32,
|
73 |
"n_vae_encoder_layers": 4,
|
74 |
-
"temporal_ensemble_coeff":
|
75 |
"dropout": 0.1,
|
76 |
"kl_weight": 10.0,
|
77 |
"optimizer_lr": 1e-05,
|
|
|
1 |
{
|
|
|
2 |
"n_obs_steps": 1,
|
3 |
"normalization_mapping": {
|
4 |
"VISUAL": "MEAN_STD",
|
|
|
55 |
},
|
56 |
"device": "cuda",
|
57 |
"use_amp": false,
|
58 |
+
"chunk_size": 300,
|
59 |
+
"n_action_steps": 1,
|
60 |
"vision_backbone": "resnet18",
|
61 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
62 |
"replace_final_stride_with_dilation": false,
|
|
|
70 |
"use_vae": true,
|
71 |
"latent_dim": 32,
|
72 |
"n_vae_encoder_layers": 4,
|
73 |
+
"temporal_ensemble_coeff": 0.01,
|
74 |
"dropout": 0.1,
|
75 |
"kl_weight": 10.0,
|
76 |
"optimizer_lr": 1e-05,
|
checkpoints/020000/pretrained_model/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:774e3f06c9cdd3663ad5029177564a81e954beec3a210ca2c5f792d3e6583b92
|
3 |
+
size 207586632
|
checkpoints/020000/pretrained_model/train_config.json
CHANGED
@@ -123,8 +123,8 @@
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
-
"chunk_size":
|
127 |
-
"n_action_steps":
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
@@ -138,7 +138,7 @@
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
-
"temporal_ensemble_coeff":
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
+
"chunk_size": 300,
|
127 |
+
"n_action_steps": 1,
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
+
"temporal_ensemble_coeff": 0.01,
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
checkpoints/020000/training_state/optimizer_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:570daaf983185fd08f186fd4002469df0e493342b705752e584ca1c748d540d9
|
3 |
+
size 413767996
|
checkpoints/020000/training_state/rng_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 15708
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:65e9bb1f9869b13fc703879a9c86300520a577ddd5f43213ad48f3829d87e871
|
3 |
size 15708
|
checkpoints/040000/pretrained_model/config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
@@ -56,8 +55,8 @@
|
|
56 |
},
|
57 |
"device": "cuda",
|
58 |
"use_amp": false,
|
59 |
-
"chunk_size":
|
60 |
-
"n_action_steps":
|
61 |
"vision_backbone": "resnet18",
|
62 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
63 |
"replace_final_stride_with_dilation": false,
|
@@ -71,7 +70,7 @@
|
|
71 |
"use_vae": true,
|
72 |
"latent_dim": 32,
|
73 |
"n_vae_encoder_layers": 4,
|
74 |
-
"temporal_ensemble_coeff":
|
75 |
"dropout": 0.1,
|
76 |
"kl_weight": 10.0,
|
77 |
"optimizer_lr": 1e-05,
|
|
|
1 |
{
|
|
|
2 |
"n_obs_steps": 1,
|
3 |
"normalization_mapping": {
|
4 |
"VISUAL": "MEAN_STD",
|
|
|
55 |
},
|
56 |
"device": "cuda",
|
57 |
"use_amp": false,
|
58 |
+
"chunk_size": 300,
|
59 |
+
"n_action_steps": 1,
|
60 |
"vision_backbone": "resnet18",
|
61 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
62 |
"replace_final_stride_with_dilation": false,
|
|
|
70 |
"use_vae": true,
|
71 |
"latent_dim": 32,
|
72 |
"n_vae_encoder_layers": 4,
|
73 |
+
"temporal_ensemble_coeff": 0.01,
|
74 |
"dropout": 0.1,
|
75 |
"kl_weight": 10.0,
|
76 |
"optimizer_lr": 1e-05,
|
checkpoints/040000/pretrained_model/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9471c811545084ebeb40c3fe154894709eaf5bf0a0021c960eef0544e06caf54
|
3 |
+
size 207586632
|
checkpoints/040000/pretrained_model/train_config.json
CHANGED
@@ -123,8 +123,8 @@
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
-
"chunk_size":
|
127 |
-
"n_action_steps":
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
@@ -138,7 +138,7 @@
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
-
"temporal_ensemble_coeff":
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
+
"chunk_size": 300,
|
127 |
+
"n_action_steps": 1,
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
+
"temporal_ensemble_coeff": 0.01,
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
checkpoints/040000/training_state/optimizer_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c6484e3108e3b2fee47fd585cd81c482b3fe9652dd9a0cfacec7fbc5e07c9d1c
|
3 |
+
size 413767996
|
checkpoints/040000/training_state/rng_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 15708
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:65288bff43e48b74558833e288ba844e2ed35b34f430710aa12e547c74751660
|
3 |
size 15708
|
checkpoints/060000/pretrained_model/config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
@@ -56,8 +55,8 @@
|
|
56 |
},
|
57 |
"device": "cuda",
|
58 |
"use_amp": false,
|
59 |
-
"chunk_size":
|
60 |
-
"n_action_steps":
|
61 |
"vision_backbone": "resnet18",
|
62 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
63 |
"replace_final_stride_with_dilation": false,
|
@@ -71,7 +70,7 @@
|
|
71 |
"use_vae": true,
|
72 |
"latent_dim": 32,
|
73 |
"n_vae_encoder_layers": 4,
|
74 |
-
"temporal_ensemble_coeff":
|
75 |
"dropout": 0.1,
|
76 |
"kl_weight": 10.0,
|
77 |
"optimizer_lr": 1e-05,
|
|
|
1 |
{
|
|
|
2 |
"n_obs_steps": 1,
|
3 |
"normalization_mapping": {
|
4 |
"VISUAL": "MEAN_STD",
|
|
|
55 |
},
|
56 |
"device": "cuda",
|
57 |
"use_amp": false,
|
58 |
+
"chunk_size": 300,
|
59 |
+
"n_action_steps": 1,
|
60 |
"vision_backbone": "resnet18",
|
61 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
62 |
"replace_final_stride_with_dilation": false,
|
|
|
70 |
"use_vae": true,
|
71 |
"latent_dim": 32,
|
72 |
"n_vae_encoder_layers": 4,
|
73 |
+
"temporal_ensemble_coeff": 0.01,
|
74 |
"dropout": 0.1,
|
75 |
"kl_weight": 10.0,
|
76 |
"optimizer_lr": 1e-05,
|
checkpoints/060000/pretrained_model/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:18472d8927d1d6ec6316a663cd56fc57f16f91f5e8352ad37265ae599cdd2d6d
|
3 |
+
size 207586632
|
checkpoints/060000/pretrained_model/train_config.json
CHANGED
@@ -123,8 +123,8 @@
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
-
"chunk_size":
|
127 |
-
"n_action_steps":
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
@@ -138,7 +138,7 @@
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
-
"temporal_ensemble_coeff":
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
+
"chunk_size": 300,
|
127 |
+
"n_action_steps": 1,
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
+
"temporal_ensemble_coeff": 0.01,
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
checkpoints/060000/training_state/optimizer_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2eeeb2338e9f6dea432f2c3ccd08a449d911105c5d21d4b6a9f4e529f550f0a2
|
3 |
+
size 413767996
|
checkpoints/060000/training_state/rng_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 15708
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f4cb3dc7e01c76bac056b07b1d73c96c8bc0cfaf0b6f26ac629203ec7c34f30d
|
3 |
size 15708
|
checkpoints/080000/pretrained_model/config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
@@ -56,8 +55,8 @@
|
|
56 |
},
|
57 |
"device": "cuda",
|
58 |
"use_amp": false,
|
59 |
-
"chunk_size":
|
60 |
-
"n_action_steps":
|
61 |
"vision_backbone": "resnet18",
|
62 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
63 |
"replace_final_stride_with_dilation": false,
|
@@ -71,7 +70,7 @@
|
|
71 |
"use_vae": true,
|
72 |
"latent_dim": 32,
|
73 |
"n_vae_encoder_layers": 4,
|
74 |
-
"temporal_ensemble_coeff":
|
75 |
"dropout": 0.1,
|
76 |
"kl_weight": 10.0,
|
77 |
"optimizer_lr": 1e-05,
|
|
|
1 |
{
|
|
|
2 |
"n_obs_steps": 1,
|
3 |
"normalization_mapping": {
|
4 |
"VISUAL": "MEAN_STD",
|
|
|
55 |
},
|
56 |
"device": "cuda",
|
57 |
"use_amp": false,
|
58 |
+
"chunk_size": 300,
|
59 |
+
"n_action_steps": 1,
|
60 |
"vision_backbone": "resnet18",
|
61 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
62 |
"replace_final_stride_with_dilation": false,
|
|
|
70 |
"use_vae": true,
|
71 |
"latent_dim": 32,
|
72 |
"n_vae_encoder_layers": 4,
|
73 |
+
"temporal_ensemble_coeff": 0.01,
|
74 |
"dropout": 0.1,
|
75 |
"kl_weight": 10.0,
|
76 |
"optimizer_lr": 1e-05,
|
checkpoints/080000/pretrained_model/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e7f9c993e99890613be1e83ca03ee1f9353224665aa535a87e83e78019aa9f1d
|
3 |
+
size 207586632
|
checkpoints/080000/pretrained_model/train_config.json
CHANGED
@@ -123,8 +123,8 @@
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
-
"chunk_size":
|
127 |
-
"n_action_steps":
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
@@ -138,7 +138,7 @@
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
-
"temporal_ensemble_coeff":
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
+
"chunk_size": 300,
|
127 |
+
"n_action_steps": 1,
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
+
"temporal_ensemble_coeff": 0.01,
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
checkpoints/080000/training_state/optimizer_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d04d60c7f98b303f95e9e22f88507a53edb598ea910ab226e5bc4537444c6ab
|
3 |
+
size 413767996
|
checkpoints/080000/training_state/rng_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 15708
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6e901fe79d08fc04f7df0d13c256dd3ea79ce08cb7f036b3c8423a3244cafe3b
|
3 |
size 15708
|
checkpoints/100000/pretrained_model/config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"type": "act",
|
3 |
"n_obs_steps": 1,
|
4 |
"normalization_mapping": {
|
5 |
"VISUAL": "MEAN_STD",
|
@@ -56,8 +55,8 @@
|
|
56 |
},
|
57 |
"device": "cuda",
|
58 |
"use_amp": false,
|
59 |
-
"chunk_size":
|
60 |
-
"n_action_steps":
|
61 |
"vision_backbone": "resnet18",
|
62 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
63 |
"replace_final_stride_with_dilation": false,
|
@@ -71,7 +70,7 @@
|
|
71 |
"use_vae": true,
|
72 |
"latent_dim": 32,
|
73 |
"n_vae_encoder_layers": 4,
|
74 |
-
"temporal_ensemble_coeff":
|
75 |
"dropout": 0.1,
|
76 |
"kl_weight": 10.0,
|
77 |
"optimizer_lr": 1e-05,
|
|
|
1 |
{
|
|
|
2 |
"n_obs_steps": 1,
|
3 |
"normalization_mapping": {
|
4 |
"VISUAL": "MEAN_STD",
|
|
|
55 |
},
|
56 |
"device": "cuda",
|
57 |
"use_amp": false,
|
58 |
+
"chunk_size": 300,
|
59 |
+
"n_action_steps": 1,
|
60 |
"vision_backbone": "resnet18",
|
61 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
62 |
"replace_final_stride_with_dilation": false,
|
|
|
70 |
"use_vae": true,
|
71 |
"latent_dim": 32,
|
72 |
"n_vae_encoder_layers": 4,
|
73 |
+
"temporal_ensemble_coeff": 0.01,
|
74 |
"dropout": 0.1,
|
75 |
"kl_weight": 10.0,
|
76 |
"optimizer_lr": 1e-05,
|
checkpoints/100000/pretrained_model/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d0e99136c7e35aac27ae1233eb041d3963aa91b394072c03298edcafa4e2f4c2
|
3 |
+
size 207586632
|
checkpoints/100000/pretrained_model/train_config.json
CHANGED
@@ -123,8 +123,8 @@
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
-
"chunk_size":
|
127 |
-
"n_action_steps":
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
@@ -138,7 +138,7 @@
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
-
"temporal_ensemble_coeff":
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
|
|
123 |
},
|
124 |
"device": "cuda",
|
125 |
"use_amp": false,
|
126 |
+
"chunk_size": 300,
|
127 |
+
"n_action_steps": 1,
|
128 |
"vision_backbone": "resnet18",
|
129 |
"pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
|
130 |
"replace_final_stride_with_dilation": false,
|
|
|
138 |
"use_vae": true,
|
139 |
"latent_dim": 32,
|
140 |
"n_vae_encoder_layers": 4,
|
141 |
+
"temporal_ensemble_coeff": 0.01,
|
142 |
"dropout": 0.1,
|
143 |
"kl_weight": 10.0,
|
144 |
"optimizer_lr": 1e-05,
|
checkpoints/100000/training_state/optimizer_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ea70a8803cdd2e43bffb073e5217016804331c1a611c50170847cd131dfce8b3
|
3 |
+
size 413767996
|
checkpoints/100000/training_state/rng_state.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 15708
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8f5a629fdddacb79efecf1465423760dea5714c1d88519c5aa9cdca84dd1d30c
|
3 |
size 15708
|