Upload folder using huggingface_hub

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1800a8cb64a8ad262a2e43c47448eef314fbee044bea704a8af95ab5ea937a91
+oid sha256:4bfb8213c476c673d1a959a34da435018569c16316e4743ad52461d655823736
 size 1066805908

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c5f65cdb47897e7a030860d390f50d38b2a8a3a39a7b971302bae18a725ab9c8
+oid sha256:6cc9821e2ba63573f7e3f0fac380d6376dcb62ff8fbd5649437db67fdcf5c978
 size 4000

policy_preprocessor_step_3_normalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c5f65cdb47897e7a030860d390f50d38b2a8a3a39a7b971302bae18a725ab9c8
+oid sha256:6cc9821e2ba63573f7e3f0fac380d6376dcb62ff8fbd5649437db67fdcf5c978
 size 4000

train_config.json

@@ -170,12 +170,12 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/output_diffusion_2",
+    "output_dir": "outputs/train/output_diffusion_3",
     "job_name": "pick_and_place",
     "resume": false,
     "seed": 1000,
-    "num_workers": 4,
-    "batch_size": 16,
+    "num_workers": 8,
+    "batch_size": 96,
     "steps": 30000,
     "eval_freq": 20000,
     "log_freq": 200,