mrinath123

import torch
import torch.nn as nn
import torch.nn.functional as F

#torch.__version__ should be 1.9
class ConvMixer(nn.Module):
    def __init__(self,no_of_op_channels , depth , kernal , patch_size ,output ):
        super().__init__()
        self.o = no_of_op_channels
        self.d = depth