Upload config.json with huggingface_hub
Browse files- config.json +3 -3
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"save_path": "
|
3 |
"model": {
|
4 |
"base_learning_rate": 8e-05,
|
5 |
"target": "latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion",
|
@@ -88,7 +88,7 @@
|
|
88 |
"data": {
|
89 |
"target": "data.data_processing.datasets.DataModule",
|
90 |
"params": {
|
91 |
-
"batch_size":
|
92 |
"num_workers": 4,
|
93 |
"wrap": false,
|
94 |
"shuffle": true,
|
@@ -99,7 +99,7 @@
|
|
99 |
"train": {
|
100 |
"target": "data.data_processing.datasets.ActionsData",
|
101 |
"params": {
|
102 |
-
"data_csv_path": "train_dataset/
|
103 |
}
|
104 |
}
|
105 |
}
|
|
|
1 |
{
|
2 |
+
"save_path": "saved_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384_cont2_ddd",
|
3 |
"model": {
|
4 |
"base_learning_rate": 8e-05,
|
5 |
"target": "latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion",
|
|
|
88 |
"data": {
|
89 |
"target": "data.data_processing.datasets.DataModule",
|
90 |
"params": {
|
91 |
+
"batch_size": 8,
|
92 |
"num_workers": 4,
|
93 |
"wrap": false,
|
94 |
"shuffle": true,
|
|
|
99 |
"train": {
|
100 |
"target": "data.data_processing.datasets.ActionsData",
|
101 |
"params": {
|
102 |
+
"data_csv_path": "train_dataset/train_dataset_20000.csv"
|
103 |
}
|
104 |
}
|
105 |
}
|