search

ethanfetaya / NRI

Neural relational inference for interacting systems - pytorch
MIT License
732 stars 157 forks source link

what does mean logits shape? #38

Closed moonnyeon closed 2 years ago

moonnyeon commented 2 years ago

what does mean logits shape ? logits = encoder(pts, rel_rec, rel_send)

my pts ----> torch.Size([32, 14, 30, 3])

logits - torch.Size([32, 182, 3])