Unet改进35:添加FastKANConv2DLayer(2024最新改进方法)
本文内容:在不同位置添加FastKANConv2DLayer
目录
论文简介
1.步骤一
2.步骤二
3.步骤三
4.步骤四
论文简介
地址
1.步骤一
新建blocks/fastkan.py文件,添加如下代码:
import torch
import torch.nn as nnclass RadialBasisFunction(nn.Module):def __init__(self,grid_min: float = -2.,grid_max: float = 2.,num_grids: int = 8,denominator: float = None, # larger denominators lead to smoother basis):super().__init__()grid