search

LymanSong / FP_GNN

A framework for detecting and classifying indoor elements in a floor plan using Graph Neural Network.
MIT License
20 stars 4 forks source link

TypeError: __init__() missing 1 required positional argument: 'device' #3

Closed taewonkang closed 3 years ago

taewonkang commented 3 years ago

Hey,

I read your paper and wanted to test the code, but the error occurs-

I tried to do the test run, and train_test.py code runs fine. But when I run test_code.py code subsequently, capture This happens.

Guess it is a simple problem, but I cannot handle since I dunno where to init 'device'... Can you help me out?

LymanSong commented 3 years ago

Hi there.

I added 'device' argument to DWGNN model in line 60

I can't test it on my desktop now, so please rerun your code and tell me if error occurs again.

Thank you for paying attention to my works!

Hope it helps.

taewonkang commented 3 years ago

Thanks, it works now :)