deepfracture / config.json
nikoloside's picture
Upload folder using huggingface_hub
b152a33 verified
{
"model_name": "pre-trained-v2",
"version": "2.0.0",
"description": "Physics-based 3D object deformation models",
"architecture": "encoder-decoder",
"framework": "pytorch",
"available_models": [
{
"name": "base",
"description": "Base object deformation model",
"training_samples": 277,
"file_size_mb": 360,
"files": [
"base.obj",
"base-encoder.pt",
"base-decoder.pt",
"base-1000-encoder.pt",
"base-1000-decoder.pt"
]
},
{
"name": "pot",
"description": "Pot object deformation model",
"training_samples": 433,
"file_size_mb": 367,
"files": [
"pot.obj",
"pot-encoder.pt",
"pot-decoder.pt",
"pot-1000-encoder.pt",
"pot-1000-decoder.pt"
]
},
{
"name": "squirrel",
"description": "Squirrel object deformation model",
"training_samples": 0,
"file_size_mb": 0,
"files": [
"squirrel.obj",
"squirrel-encoder.pt",
"squirrel-decoder.pt",
"squirrel-1000-encoder.pt",
"squirrel-1000-decoder.pt"
]
},
{
"name": "bunny",
"description": "Bunny object deformation model",
"training_samples": 0,
"file_size_mb": 0,
"files": [
"bunny.obj",
"bunny-encoder.pt",
"bunny-decoder.pt",
"bunny-1000-encoder.pt",
"bunny-1000-decoder.pt"
]
},
{
"name": "lion",
"description": "Lion object deformation model",
"training_samples": 0,
"file_size_mb": 0,
"files": [
"lion.obj",
"lion-encoder.pt",
"lion-decoder.pt",
"lion-1000-encoder.pt",
"lion-1000-decoder.pt"
]
}
],
"model_config": {
"encoder": {
"type": "mlp_with_attention",
"input_dim": 9,
"hidden_dims": [512, 256, 128],
"output_dim": 64
},
"decoder": {
"type": "geometric_reconstruction",
"input_dim": 64,
"hidden_dims": [128, 256, 512],
"output_dim": 3
},
"training": {
"optimizer": "adam",
"learning_rate": 1e-4,
"batch_size": 32,
"epochs": 1000,
"loss_function": "combined_geometric_physics"
}
},
"requirements": {
"python": ">=3.8",
"pytorch": ">=1.9.0",
"numpy": ">=1.21.0",
"trimesh": ">=3.9.0"
},
"license": "Apache-2.0",
"authors": ["Your Name"],
"contact": "your.email@example.com"
}