Upload checkpoint-550000 to giorom-3d-t-sand3d-long

giorom-3d-t-sand3d-long/config.json

@@ -0,0 +1,57 @@
+{
+  "_name_or_path": "/scratch/gilbreth/hviswan/GIOROM/saved_models/giorom3d-t-sand3d",
+  "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
+}

giorom-3d-t-sand3d-long/model.safetensors

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

giorom-3d-t-sand3d-long/optimizer.pt

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

giorom-3d-t-sand3d-long/rng_state.pth

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

giorom-3d-t-sand3d-long/scheduler.pt

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

giorom-3d-t-sand3d-long/trainer_state.json

@@ -0,0 +1,41 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 77.94784580498866,
+  "eval_steps": 500,
+  "global_step": 550000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "avg_loss": 0.00010567396622950003,
+      "grad_norm": 0.6266661882400513,
+      "learning_rate": 4.22180373712577e-05,
+      "loss": 1.8564207493909635e-05,
+      "lr": 4.21791904474444e-05,
+      "step": 549999
+    }
+  ],
+  "logging_steps": 1500,
+  "max_steps": 14112000,
+  "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
+}

giorom-3d-t-sand3d-long/training_args.bin

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