refactor/pretrainedconfig_device_amps_args (#3)

- Upload folder using huggingface_hub (e7bf12d26c4d50a9386ff209b643d24f7009b61d)
- fix/add_args_to_nested_policy_config (6757eddf48d727c4f297581dbe62f17b9fdd6ae3)


Co-authored-by: Steven Palma <[email protected]>

config.json

@@ -38,6 +38,8 @@
   "n_action_steps": 8,
   "n_groups": 8,
   "n_obs_steps": 2,
+  "device": "cuda",
+  "use_amp": false,
   "noise_scheduler_type": "DDPM",
   "normalization_mapping": {
     "ACTION": "MIN_MAX",

train_config.json

@@ -104,6 +104,8 @@
     "policy": {
         "type": "diffusion",
         "n_obs_steps": 2,
+        "device": "cuda",
+        "use_amp": false,
         "normalization_mapping": {
             "VISUAL": "MEAN_STD",
             "STATE": "MIN_MAX",
@@ -178,8 +180,6 @@
     "output_dir": "outputs/train/2025-01-17/11-51-15_pusht_diffusion",
     "job_name": "pusht_diffusion",
     "resume": false,
-    "device": "cuda",
-    "use_amp": false,
     "seed": 100000,
     "num_workers": 4,
     "batch_size": 64,