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):