hrishivish23's picture
Upload checkpoint-220000 to giorom-3d-t-sand3d
fa6f5d0 verified
{
"_name_or_path": "/scratch/gilbreth/hviswan/GIOROM/saved_models/giorom-3d-t-elasticity",
"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
}