search

KaihuaTang / VCTree-Scene-Graph-Generation

Code for the Scene Graph Generation part of CVPR 2019 oral paper: "Learning to Compose Dynamic Tree Structures for Visual Contexts"
MIT License
121 stars 15 forks source link

assert all(map(lambda i: i.is_cuda, inputs)) #2

Open HCPLab-SYSU opened 5 years ago

HCPLab-SYSU commented 5 years ago

When I use multi-gpus, the error "assert all(map(lambda i: i.is_cuda, inputs))" appears. If anyone knows it, please tell me why. Thanks!