调G2G论文代码之流水账

流水账记录

运行代码前需要改一些导入,因为作者代码下载后,一些不同文件夹中的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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值