there is no MLPEncoder class

PhyllisH / NMP

Code for Neural Message Passing for Visual Relationship Detection

31 stars 8 forks source link

there is no MLPEncoder class #5

Open xywhat opened 2 years ago

xywhat commented 2 years ago

class NMPEncoder(nn.Module): def init(self, n_hid, edge_types=71, node_types=101, n_iter=2, do_prob=0., use_vis=True, use_spatial=False, use_sem=True, use_loc=False, use_cls=False): super(MLPEncoder, self).init()