流水账记录
运行代码前需要改一些导入,因为作者代码下载后,一些不同文件夹中的PY文件导入出现问题。
比如from fast_jtnn import *会报无fast_jtnn 模块
需要添加代码如下:(也就是手动把文件夹下的所有函数导入一遍)其他情况同理
import os
import sys
sys.path.append(os.path.abspath('../fast_jtnn'))
from mol_tree import Vocab, MolTree
from diff_vae import DiffVAE
from scaff_gan import ScaffoldGAN
from jtnn_enc import JTNNEncoder
from jtmpn import JTMPN
from mpn import MPN
from nnutils import create_var
from datautils import MolTreeFolder, PairTreeFolder, MolTreeDataset
运行G2G论文里的代码时发现报警 UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
这是pytorch版本更换以后导致的警告,(我按照作者说的安装的torch0.4.1,估计作者运行也是这样子把)看样子不影响运行,但程序也不动了,不知道是不是因为它的情况下只能修改代码了。
所有用到size_average 和reduce 的地方都需要换成reduction='sum'
经过查找发现3处需要更改的地方,都在fast_jtnn下,好吧,改起。jtnn_dec.py和diff_vae.py