priba / nmp_qc Goto Github PK
View Code? Open in Web Editor NEWOur own implementation of Neural Message Passing for Computer Vision paper
License: MIT License
Our own implementation of Neural Message Passing for Computer Vision paper
License: MIT License
In the original paper, the readout function is different between ggcn and the function used in the paper. But I noticed that you implement them as the same function.
Have I missed something?
Hi, I am trying to run the program on Ubuntu 16.04 with python 2.7. I have the rdkit
package installed and the data downloaded. However, when I tried to run the main program I got the following error:
(venv) siyuan:nmp_qc$ python main.py
Prepare files
Define model
python: /usr/include/boost/thread/pthread/mutex.hpp:111: boost::mutex::~mutex(): Assertion `!res' failed.
Aborted (core dumped)
The program crashed here:
Lines 101 to 103 in 177db7e
What would possibly be the reason for that? Thanks in advance!
Hello, thank you for your implemenation!
I've just tried to run default experiment with
python main.py --no-cuda --epochs 1
and run into the following problem
/opt/conda/lib/python3.6/importlib/_bootstrap.py:205: RuntimeWarning: compiletime version 3.5 of module 'tensorflow.python.framework.fast_tensor_util' does not match runtime version 3.6 return f(*args, **kwds)
Prepare files
Define model
Statistics
Create model
Optimizer
Logger
=> no best model found at './checkpoint/qm9/mpnn/model_best.pth'
Check cuda
Traceback (most recent call last):
File "main.py", line 321, in <module>
main()
File "main.py", line 182, in main
train(train_loader, model, criterion, optimizer, epoch, evaluation, logger)
File "main.py", line 242, in train
output = model(g, h, e)
File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 319, in __call__
result = self.forward(*input, **kwargs)
File "/data/grishin/nmp_qc/models/MPNN.py", line 78, in forward
m = self.m[0].forward(h[t], h_aux, e_aux)
File "/data/grishin/nmp_qc/MessageFunction.py", line 43, in forward
return self.m_function(h_v, h_w, e_vw, args)
File "/data/grishin/nmp_qc/MessageFunction.py", line 175, in m_mpnn
h_w_rows = h_w[..., None].expand(h_w.size(0), h_v.size(1), h_w.size(1)).contiguous()
RuntimeError: The expanded size of the tensor (25) must match the existing size (73) at non-singleton dimension 1
Am i doing something wrong? Thank you in advance.
Why does the project title has Computer Vision in it? The paper that the project implements is on Quantum Chemistry.
When I followed the installation that installed rdkit and all other requirements, and downloaded the data, run the "python main.py", I got the error below. Does there something error in the code?
Prepare files
Define model
Traceback (most recent call last):
File "/home/ay27/repo/nmp_qc/main.py", line 320, in <module>
main()
File "/home/ay27/repo/nmp_qc/main.py", line 103, in main
g_tuple, l = data_train[0]
File "/home/ay27/repo/nmp_qc/datasets/qm9.py", line 47, in __getitem__
h = self.vertex_transform(g)
File "/home/ay27/repo/nmp_qc/datasets/utils.py", line 28, in qm9_nodes
for n, d in g.nodes_iter(data=True):
AttributeError: 'Graph' object has no attribute 'nodes_iter'
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.