JIT-compilation fails
Thank you for this great library, I recently I tried out the new JIT-fusion mode however it fails with this message:
```
tiny-cuda-nn warning: duplicate named arg
Failed to JIT-compile `training_step_network_with_input_encoding`. Disabling JIT.
```
My input size is 14, my output size is 3, my batch size 256 * 64 * 8 and my configuration:
```json
{
"encoding": {
"otype": "Composite",
"nested": [
{
"n_dims_to_encode": 3,
"otype": "Grid",
"type": "Hash",
"n_levels": 16,
"n_features_per_level": 2,
"log2_hashmap_size": 19,
"base_resolution": 16,
"per_level_scale": 2.0,
"interpolation": "Linear"
},
{
"n_dims_to_encode": 5,
"otype": "OneBlob",
"n_bins": 4
},
{
"otype": "Identity"
}
]
},
"network": {
"otype": "FullyFusedMLP",
"activation": "ReLU",
"output_activation": "None",
"n_neurons": 64,
"n_hidden_layers": 2
},
"loss": {
"otype": "RelativeL2Luminance"
},
"optimizer": {
"otype": "EMA",
"decay": 0.95,
"nested": {
"otype": "Adam",
"learning_rate": 1e-2,
"beta1": 0.9,
"beta2": 0.99,
"epsilon": 1e-15,
"l2_reg": 1e-6,
"absolute_decay": 0,
"relative_decay": 0,
"adabound": false
}
}
}
```
Does someone know what could be going wrong here? I run my tests on C++20 using GCC14, CudaToolkit 12.8.93, Fedora Linux and an RTX 3060 Ti.
Any help appreciated
1 条评论