yuntian-deng commited on
Commit
242e295
·
verified ·
1 Parent(s): e40cf54

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "save_path": "saved_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384_cont2",
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": 64,
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/train_dataset.csv"
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
  }