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 24fd7d2 commit 7a75178Copy full SHA for 7a75178
timm/models/csatv2.py
@@ -147,7 +147,7 @@ def __init__(
147
super().__init__()
148
kernel = {'2': _dct_kernel_type_2, '3': _dct_kernel_type_3}
149
dct_weights = kernel[f'{kernel_type}'](kernel_size, orthonormal, **dd).T
150
- self.register_buffer('weights', dct_weights)
+ self.register_buffer('weights', dct_weights.contiguous())
151
self.register_parameter('bias', None)
152
153
def forward(self, x: torch.Tensor) -> torch.Tensor:
@@ -705,7 +705,7 @@ def _cfg(url='', **kwargs):
705
706
default_cfgs = generate_default_cfgs({
707
'csatv2': _cfg(
708
- url='https://huggingface.co/Hyunil/CSATv2/resolve/main/CSATv2_ImageNet_timm.pth'
+ hf_hub_id='timm/',
709
),
710
'csatv2_21m': _cfg(),
711
})
0 commit comments