dependency fix

main
Vladimir Protsenko 7 months ago
parent 7d28a70226
commit ab71a512de

@ -5,7 +5,6 @@ import numpy as np
from common.utils import round_func from common.utils import round_func
from common.lut import forward_rc_conv_centered, forward_rc_conv_rot90, forward_2x2_input_SxS_output from common.lut import forward_rc_conv_centered, forward_rc_conv_rot90, forward_2x2_input_SxS_output
from pathlib import Path from pathlib import Path
from einops import repeat
class RCLutCentered_3x3(nn.Module): class RCLutCentered_3x3(nn.Module):
def __init__( def __init__(

Loading…
Cancel
Save