Original file line numberDiff line numberDiff line change @@ -0,0 +1,43 @@ from typing import List import torch import torch.nn as nn class MatryoshkaHead(nn.Module): """ https://github.com/RAIVNLab/MRL/blob/main/MRL.py Matryoshka linear layer for Matryoshka Representation Learning ""...
Time step: 259, since reset: 56 Traceback (most recent call last): File ".../tmrl/tmrl/tuto/tuto.py", line 535, in <module> run_trainer(my_trainer) File ".../tmrl/tmrl/tuto/tuto.py", line 528, in run_trainer trainer.run() File ".../tmrl/tmrl/networking.py", line ...
Module): def __init__( self, layer_sizes, num_classes: int = 10, input_channels: int = 1, spline_order: int = 3, groups: int = 1): super(SimpleConvKAN, self).__init__() self.layers = nn.Sequential( KANConv2DLayer(input_channels, layer_sizes[0], spline_order, kernel_size=...
importtorchimporttorch.nnasnnfromkan_convsimportKANConv2DLayerclassSimpleConvKAN(nn.Module):def__init__(self,layer_sizes,num_classes:int=10,input_channels:int=1,spline_order:int=3,groups:int=1):super(SimpleConvKAN,self).__init__()self.layers=nn.Sequential(KANConv2DLayer(input_channels,laye...