ckpts/neural_basis_3layer_relu_fp16.json

{
    "name": "NeuralBasis",
    "args":{
      "input_dim": 32,
      "output_dim": 5,
      "hidden_dim": 64,
      "num_layers": 3,
      "use_normal": true,
      "activation": "relu",
      "normal_dim": 3,
      "num_frequencies": 4,
      "use_residual": false
    }
}