English
wala
unconditional-3d-generation
WaLa-UN-1B / args.json
kamalrahimi's picture
Create args.json
655bc25 verified
{
"strategy": null,
"gpu_workers": 8,
"restore_path": null,
"prefetch_factor": null,
"matual_precision": "medium",
"max_concurrency": 1500,
"multipart_size": 8388608,
"print_every": 100,
"save_every": 100,
"validation_every": 15000,
"validation_every_log": 500,
"visualization_every": 15000,
"log_level": "info",
"experiment_type": "max",
"experiment_every": 5,
"fast_dev_run": false,
"limit_val_batches": 0.05,
"filter_path": null,
"finetune": false,
"finetune_dp_cond": null,
"finetune_dp_cond_type": null,
"ft_train_number": null,
"ft_train_datasets": null,
"val_cnt": null,
"use_even_val": false,
"use_ema": true,
"ema_decay": 0.9999,
"use_ema_weights": true,
"use_compile": true,
"batch_size": 8,
"test_batch_size": 8,
"num_workers": 16,
"test_threshold": 0.05,
"num_gpus": 8,
"seed": 1,
"epochs": 300,
"optimizer": "Adam",
"lr": 0.0001,
"train_mode": "train",
"use_local_storage": false,
"auto_exp_name": null,
"input_type": "Wavelet",
"output_type": "Wavelet",
"encoder_type": "General_Encoder_Down_2",
"decoder_type": "General_Decoder_Up_2",
"encoder_num_tran": 0,
"decoder_num_tran": 0,
"last_feature_transform": null,
"reconstruct_loss_type": "mean",
"quantizer_type": "original",
"normalize_latent": null,
"e_dim": 4,
"n_e": 1024,
"beta": 0.25,
"sample_mode": "bilinear",
"padding": 0.1,
"gamma": 1,
"grid_size": 12,
"t_loss": 1.0,
"num_latent_tokens": 256,
"dataset_name": "all",
"voxel_transform": null,
"num_points": 2048,
"num_sdf_points": 5000,
"categories": null,
"resolution": 256,
"max_depth": 3,
"max_training_level": 2,
"point_num": 16384,
"keep_level": 2,
"data_keep_level": 2,
"wavelet": "bior6.8",
"padding_mode": "constant",
"use_normalization": false,
"use_shift_mean": false,
"start_stage": 0,
"use_adaptive_stage_update": false,
"no_rebalance_loss": true,
"use_compact_indices": true,
"sample_threshold_ratio": 0.03125,
"use_batched_threshold": true,
"use_sample_training": true,
"use_sample_threshold": true,
"div_hyp": 1.0,
"checkpoint": null,
"use_timestamp": false,
"num_iterations": 300000,
"gpu": "0",
"threshold": 0.45,
"sampling_type": null,
"auto_precision": "bf16",
"gradient_clip_val": 1.0,
"dropout": 0.0,
"sdf_points": 20000,
"sdf_sample_type": "mixture",
"sdf_res": 256,
"greater_or_no": true,
"wavelet_transform": "all",
"test_exp_name": "default",
"test_file_name": "model_performance.csv",
"network_type": "latent_uvit",
"diffusion_beta_schedule": "cosine",
"diffusion_step": 1000,
"diffusion_rescale_timestep": 1000,
"diffusion_scale_ratio": 1.0,
"diffusion_model_var_type": "FIXED_SMALL",
"diffusion_model_mean_type": "START_X",
"diffusion_loss_type": "MSE",
"diffusion_sampler": "second-order",
"dit_block_type": "dit",
"att_patch_size": 1,
"att_hidden_size": 1152,
"transformer_num_blocks": 44,
"transformer_num_heads": 16,
"transformer_add_num_register": 0,
"unet_model_channels": 128,
"unet_num_res_blocks": [
3
],
"learnable_skip_r": null,
"add_condition_res_ch": null,
"with_fix_pos": true,
"cond_num_mapping_layers": 0,
"add_condition_time_ch": true,
"add_condition_input_ch": null,
"use_pointcloud_conditions": false,
"pc_encoder_type": "PointNet_Simple",
"use_pointvoxel_encoder": false,
"num_pc_points": 2500,
"use_pc_samples": false,
"sample_num": 2500,
"pc_dims": 1024,
"num_inds": 1024,
"pc_output_dim": 1024,
"use_voxel_conditions": false,
"voxel_context_dim": 1024,
"voxel_dim": 8,
"voxel_resolution": 16,
"use_image_conditions": false,
"use_camera_index": false,
"render_resolution": 384,
"max_images_num": 55,
"image_transform": "dino",
"clip_model_type": "dino-l-14_reg",
"input_view_cnt": 4,
"use_multiple_views_grids": false,
"training_views": null,
"testing_views": null,
"use_depth_conditions": false,
"use_wavelet_conditions": false,
"dp_cond": null,
"scale": 3,
"guidance_type": null,
"dp_cond_type": null,
"use_autoencoder": false,
"checkpoint_type": "last_ft_10000",
"pre_quant": true,
"latent_normalization": false,
"use_autoencoder_ema": false,
"test_log_num": null,
"precision": "bf16",
"gradient_clip_val_2": 1.0,
"wavelet_transform_2": null,
"weight_decay": 0.0,
"opt_eps": 1e-08,
"latent_lr": 0.0001,
"pin_memory": false
}