mirror of
https://github.com/microsoft/TRELLIS.2
synced 2026-04-25 17:15:37 +02:00
134 lines
4.0 KiB
JSON
Executable File
134 lines
4.0 KiB
JSON
Executable File
{
|
|
"models": {
|
|
"encoder": {
|
|
"name": "SparseUnetVaeEncoder",
|
|
"args": {
|
|
"in_channels": 6,
|
|
"model_channels": [64, 128, 256, 512, 1024],
|
|
"latent_channels": 32,
|
|
"num_blocks": [0, 4, 8, 16, 4],
|
|
"block_type": [
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d"
|
|
],
|
|
"down_block_type": [
|
|
"SparseResBlockS2C3d",
|
|
"SparseResBlockS2C3d",
|
|
"SparseResBlockS2C3d",
|
|
"SparseResBlockS2C3d"
|
|
],
|
|
"block_args": [
|
|
{
|
|
"use_checkpoint": true
|
|
},
|
|
{
|
|
"use_checkpoint": true
|
|
},
|
|
{
|
|
"use_checkpoint": false
|
|
},
|
|
{
|
|
"use_checkpoint": false
|
|
},
|
|
{
|
|
"use_checkpoint": false
|
|
}
|
|
],
|
|
"use_fp16": true
|
|
}
|
|
},
|
|
"decoder": {
|
|
"name": "SparseUnetVaeDecoder",
|
|
"args": {
|
|
"out_channels": 6,
|
|
"model_channels": [1024, 512, 256, 128, 64],
|
|
"latent_channels": 32,
|
|
"num_blocks": [4, 16, 8, 4, 0],
|
|
"block_type": [
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d",
|
|
"SparseConvNeXtBlock3d"
|
|
],
|
|
"up_block_type": [
|
|
"SparseResBlockC2S3d",
|
|
"SparseResBlockC2S3d",
|
|
"SparseResBlockC2S3d",
|
|
"SparseResBlockC2S3d"
|
|
],
|
|
"block_args": [
|
|
{
|
|
"use_checkpoint": false
|
|
},
|
|
{
|
|
"use_checkpoint": false
|
|
},
|
|
{
|
|
"use_checkpoint": false
|
|
},
|
|
{
|
|
"use_checkpoint": true
|
|
},
|
|
{
|
|
"use_checkpoint": true
|
|
}
|
|
],
|
|
"use_fp16": true,
|
|
"pred_subdiv": false
|
|
}
|
|
}
|
|
},
|
|
"dataset": {
|
|
"name": "SparseVoxelPbrDataset",
|
|
"args": {
|
|
"resolution": 256,
|
|
"min_aesthetic_score": 4.5,
|
|
"max_active_voxels": 1000000,
|
|
"max_num_faces": 1000000,
|
|
"with_mesh": false,
|
|
"attrs": [
|
|
"base_color",
|
|
"metallic",
|
|
"roughness",
|
|
"alpha"
|
|
]
|
|
}
|
|
},
|
|
"trainer": {
|
|
"name": "PbrVaeTrainer",
|
|
"args": {
|
|
"max_steps": 1000000,
|
|
"batch_size_per_gpu": 8,
|
|
"batch_split": 1,
|
|
"optimizer": {
|
|
"name": "AdamW",
|
|
"args": {
|
|
"lr": 1e-4,
|
|
"weight_decay": 0.0
|
|
}
|
|
},
|
|
"ema_rate": [
|
|
0.9999
|
|
],
|
|
"fp16_mode": "inflat_all",
|
|
"fp16_scale_growth": 0.001,
|
|
"grad_clip": {
|
|
"name": "AdaptiveGradClipper",
|
|
"args": {
|
|
"max_norm": 1.0,
|
|
"clip_percentile": 95
|
|
}
|
|
},
|
|
"i_log": 500,
|
|
"i_sample": 10000,
|
|
"i_save": 10000,
|
|
"lambda_kl": 1e-6,
|
|
"loss_type": "l1",
|
|
"lambda_render": 0.0
|
|
}
|
|
}
|
|
} |