|
{ |
|
"_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 |
|
} |
|
|