search

PJLab-ADG / neuralsim

neuralsim: 3D surface reconstruction and simulation based on 3D neural rendering.
MIT License
582 stars 31 forks source link

Change the capacity of the model #39

Open sonnefred opened 9 months ago

sonnefred commented 9 months ago

Hi, thanks for your great work! I'm trying to do some experiments and I'd like to change the size of the model, such as the depth or width of MLP. Could you tell me where I can change it directly? Is there any hyper-parameter to control this? Thanks.

Geniussh commented 8 months ago

Look for D and W in the yaml file. For example, to change the depth and width of the radiance head, you can change assetbank_cfg.Street.model_params.radiance_cfg.D and assetbank_cfg.Street.model_params.radiance_cfg.W