fix: for CUDA version >= 12.6

This commit is contained in:
kiennt
2025-08-16 09:57:17 +00:00
parent e1420f9335
commit 546f444c1c
10 changed files with 1453 additions and 23 deletions

View File

@@ -8,7 +8,7 @@
import torch
import torch.nn as nn
import torch.nn.functional as F
from timm.models.layers import DropPath
from timm.layers import DropPath
class FeatureResizer(nn.Module):