prune_issue

[yuyijie1995]

class ModuleConnect(object):
def init(self, layer_type, layer_name, batchnorm=None, activation=None,
top=None, bottom=None, top_idx=None, bottom_idx=None):
print('now module connect init!!!')
self.layer_type = layer_type
self.layer_name = layer_name
self.batchnorm = batchnorm
self.activation = activation
self.top = top
self.top_idx = top_idx
self.bottom = bottom
self.bottom_idx = bottom_idx
self.next_mod = []
self.prev_mod=[]

def __repr__(self):#打印一个实例化对象
    print('repr module connect')
    trptd  = '-------------------------------\n'
    trptd += f'layer_type: {self.layer_type}\n'
    trptd += f'layer_name: {self.layer_name}\n'
    trptd += f'batchnorm: {self.batchnorm}\n'
    trptd += f'activation: {self.activation}\n'
    trptd += f'bottom: {self.bottom}({self.bottom_idx})\n'
    trptd += f'top: {self.top}({self.top_idx})\n'
    trptd += '-------------------------------\n'
    return trptd