We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 99dad24 commit 06cc355Copy full SHA for 06cc355
1 file changed
training/variance_task.py
@@ -124,7 +124,7 @@ def __init__(self):
124
# NOTE: Variance model backbones use num_channels=384/512, which are too
125
# small for Triton fusion to benefit. The compilation overhead outweighs
126
# the HBM savings. Fused kernels for acoustic model only (K=1024).
127
- if hparams.get('use_fused_kernels', False):
+ if hparams.get('use_fused_kernels_variance', False):
128
from modules.kernels.integration import patch_variance_model
129
from lightning.pytorch.utilities.rank_zero import rank_zero_info
130
n = patch_variance_model(
0 commit comments