search

atomistic-machine-learning / schnetpack

SchNetPack - Deep Neural Networks for Atomistic Systems
Other
791 stars 215 forks source link

Training the model on QM9. #620

Closed renatofrazzato closed 8 months ago

renatofrazzato commented 8 months ago

When I try to train the model on QM9 I got an error: ValueError: dictionary update sequence element #0 has length 1; 2 is required. When I try to run the line: trainer.fit(task, datamodule=qm9data)

I am using the exactly the same code from the tutorial notebook which was working some months ago.

Thanks in advance.

jnsLs commented 8 months ago

Dear @renatofrazzato, are you talking about that tutorial: tutorial_02_qm9 ? For me it runs without problems. You could try to delete the qm9.db file and the split.npz file from the previous runs and run the tutorials again. If that does not solve your issue, I would need more details, to help you solve it. Best, Jonas

renatofrazzato commented 8 months ago

Dear @jnsLs., thank you for your support. Yes, I was talking about tutorial_02_qm9 and I am using google colab to run the code. I could run the code again, I changed the way to install the SchnetPack I was installing the library using the command "!pip install git+https://github.com/atomistic-machine-learning/schnetpack.git" and now I tried simply "!pip install schnetpack" and it worked.

Thanks, Renato