Upload folder using huggingface_hub
Browse files- args.json +23 -0
- epoch_10.pth +3 -0
- epoch_15.pth +3 -0
- epoch_20.pth +3 -0
- epoch_25.pth +3 -0
- epoch_30.pth +3 -0
- epoch_35.pth +3 -0
- epoch_5.pth +3 -0
- latest.pth +3 -0
args.json
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"pretrained_model": "./experiments/baseline/checkpoint/latest.pth",
|
| 3 |
+
"args_file": "./experiments/baseline/checkpoint/args.json",
|
| 4 |
+
"train_set": "./preprocessed_data",
|
| 5 |
+
"train_set_list": "/home/Humble/Humble/EndtoEnd/YSH-Diffusion-Planner/diffusion_planner_training.json",
|
| 6 |
+
"num_samples": 8,
|
| 7 |
+
"beta": 1.0,
|
| 8 |
+
"weight_clip": 10.0,
|
| 9 |
+
"temperature": 1.0,
|
| 10 |
+
"learning_rate": 1e-05,
|
| 11 |
+
"gradient_accumulation_steps": 1,
|
| 12 |
+
"reward_type": "base",
|
| 13 |
+
"epochs": 50,
|
| 14 |
+
"batch_size": 256,
|
| 15 |
+
"num_workers": 16,
|
| 16 |
+
"save_freq": 5,
|
| 17 |
+
"output_dir": "./wr_output",
|
| 18 |
+
"name": "wr-base",
|
| 19 |
+
"use_mlflow": true,
|
| 20 |
+
"mlflow_tracking_uri": "https://dagshub.com/YSH-research/YSH-Diffusion-Planner.mlflow",
|
| 21 |
+
"mlflow_experiment_name": "wr-finetuning",
|
| 22 |
+
"local_rank": -1
|
| 23 |
+
}
|
epoch_10.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4eedfa3b8d864983666c50033d825638d8bc31461886481c42ec585c24dad1ad
|
| 3 |
+
size 72861851
|
epoch_15.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:792c85c85597f975b9896abcb1e262a8cf5a0e235d8b2210850957a7fa3991d0
|
| 3 |
+
size 72861851
|
epoch_20.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b0f705df6751f6727821699d95a5325eccc2dcd8ec6bb96cc0bc8e2c682d5a2
|
| 3 |
+
size 72861851
|
epoch_25.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8a5f93c7b0700b4e55c9cb1af09639d0d6002b0cee06f05828bf92d998b6dc83
|
| 3 |
+
size 72861851
|
epoch_30.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e4a1afb75d10bf66d340e5dae07730e773a311dd378c23725f99cce51b0d461
|
| 3 |
+
size 72861851
|
epoch_35.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:441e6e6b2578e0b61d712ef840169f7893875d636ea8c2a7a2def7c64963a772
|
| 3 |
+
size 72861851
|
epoch_5.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c84be345f1432d06ff32787a463abc32b24575dcbacc9c32de14d83946a76f28
|
| 3 |
+
size 72860297
|
latest.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d8bb63cf8e437e63698b38ded09bc6a8906188ba99b35fc1f43abe8943542401
|
| 3 |
+
size 72854839
|