From 6d1ffb551b554091a21ead45b4bf42f3af8beac4 Mon Sep 17 00:00:00 2001 From: Vladimir Protsenko Date: Sun, 28 Apr 2024 13:40:57 +0000 Subject: [PATCH] implemented SDYLutCenteredx1 --- src/common/layers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/common/layers.py b/src/common/layers.py index f1bc405..62a0bd3 100644 --- a/src/common/layers.py +++ b/src/common/layers.py @@ -2,6 +2,7 @@ import torch import torch.nn as nn import torch.nn.functional as F import numpy as np +from .utils import round_func class PercievePattern(): def __init__(self, receptive_field_idxes=[[0,0],[0,1],[1,0],[1,1]], center=[0,0], window_size=2):