Upload checkpoint-220000 to giorom-3d-t-sand3d

config.json

@@ -0,0 +1,57 @@
+{
+  "_name_or_path": "/scratch/gilbreth/hviswan/GIOROM/saved_models/checkpoint-220000",
+  "architectures": [
+    "PhysicsEngine"
+  ],
+  "dim": 3,
+  "gno_radius": 0.125,
+  "heads": 3,
+  "hidden_size": 128,
+  "in_gno_mlp_hidden_layers": [
+    131,
+    32,
+    64,
+    64
+  ],
+  "in_gno_transform_type": "nonlinear_kernelonly",
+  "latent_domain_lims": [
+    [
+      0.0,
+      1.0
+    ],
+    [
+      0.0,
+      1.0
+    ],
+    [
+      0.0,
+      1.0
+    ]
+  ],
+  "latent_grid_dim": 16,
+  "n_mp_layers": 2,
+  "not_branch_size": 3,
+  "not_heads": 4,
+  "not_layers": 1,
+  "not_output_size": 128,
+  "not_space_dim": 64,
+  "not_trunk_size": 64,
+  "num_particle_types": 9,
+  "out_gno_hidden": 128,
+  "out_gno_in_dim": 3,
+  "out_gno_mlp_hidden_layers": [
+    3,
+    32,
+    64,
+    128
+  ],
+  "out_gno_transform_type": "linear",
+  "particle_type_dim": 16,
+  "projection_channels": 256,
+  "projection_layers": 1,
+  "projection_n_dim": 1,
+  "torch_dtype": "float32",
+  "transformers_version": "4.49.0",
+  "use_open3d": false,
+  "window_size": 5
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5988486eae82102d817268d65bf13e2c75d158ced00daf4994b1d476e602a52
+size 9568276

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1eb9785441eb6839ef6303b6d399c399b97af4f9cab8dabb543ae2a7969a29e1
+size 19244538

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:389769e79f51e0631cb8227739225e60ff3aa16e83c5a4096cd66c07328b42af
+size 14244

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56f191a7c9a54597391a1acc630adbf0d4ab538406bf0edc4f28d4e9cfd8f5f8
+size 1000

trainer_state.json

@@ -0,0 +1,41 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 11.67728237791932,
+  "eval_steps": 500,
+  "global_step": 220000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "avg_loss": 0.29662682761985026,
+      "grad_norm": 32.271949768066406,
+      "learning_rate": 6.674212038333604e-05,
+      "loss": 0.06960329413414001,
+      "lr": 6.668070762644054e-05,
+      "step": 219999
+    }
+  ],
+  "logging_steps": 1500,
+  "max_steps": 37680000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2000,
+  "save_steps": 110000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b2d5fb1239f85339ec2dac852116c36477743f93287a6bbff77e3c3c2e6e1d0
+size 5240