motefly / DeepGBM

SIGKDD'2019: DeepGBM: A Deep Learning Framework Distilled by GBDT for Online Prediction Tasks
647 stars 135 forks source link

when I use flight dataset, I run the main.py and show follow error #9

Closed fengweisi closed 5 years ago

fengweisi commented 5 years ago

image

motefly commented 5 years ago

I didn't meet it, and could you please give more detail description about it?

fengweisi commented 5 years ago

I didn't meet it, and could you please give more detail description about it?

thank you answer,I fix it yesterday.Cause this error,because Torchvision's version is not working.when I install 0.2.0,the program can run.