From ab71a512de02fc0954d8465edd8a48b4bd22ab0a Mon Sep 17 00:00:00 2001 From: Vladimir Protsenko Date: Mon, 22 Apr 2024 10:10:53 +0000 Subject: [PATCH] dependency fix --- src/models/rclut.py | 1 - 1 file changed, 1 deletion(-) 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__(