/TreeNet

TreeNet

Primary LanguagePython

直接运行code中的train.py,eval暂时无法运行

需要的python版本:3.7
运行需要的扩展包:

Numpy,
Pandas,
Pytorch 1.2

Pytorch的安装方法:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
需要运行库:Cuda 10.0

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch
需要运行库: CUDA 9.2