search

wenhao-gao / SynNet

MIT License
84 stars 21 forks source link

Issue running optimize_ga.py #6

Closed tkram01 closed 2 years ago

tkram01 commented 2 years ago

When I try and run optimize_ga.py I am getting

Using backend: pytorch
Downloading gin_supervised_contextpred_pre_trained.pth from https://data.dgl.ai/dgllife/pre_trained/gin_supervised_contextpred.pth...
Pretrained model loaded
Starting with 128 fps with 4096 bits
Traceback (most recent call last):
  File "/home/ec2-user/SynNet/scripts/optimize_ga.py", line 205, in <module>
    scores, mols, trees = fitness(embs=population,
TypeError: fitness() got an unexpected keyword argument 'pool'

This is the command I am using 

python optimize_ga.py --radius 2 --nbits 4096 --num_population 128 --num_offspring 512 --num_gen 200 --ncpu 48 --objective logp