diff --git a/src/models/rclut.py b/src/models/rclut.py index 4cf4b06..61d780b 100644 --- a/src/models/rclut.py +++ b/src/models/rclut.py @@ -5,7 +5,6 @@ import numpy as np from common.utils import round_func from common.lut import forward_rc_conv_centered, forward_rc_conv_rot90, forward_2x2_input_SxS_output from pathlib import Path -from einops import repeat class RCLutCentered_3x3(nn.Module): def __init__(