Giters

PhyllisH / NMP

Code for Neural Message Passing for Visual Relationship Detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

there is no MLPEncoder class

xywhat opened this issue · comments

commented

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()