DAGCN icon indicating copy to clipboard operation
DAGCN copied to clipboard

Which version of torch do u use in the project?

Open sunfeng90 opened this issue 2 years ago • 0 comments

Thanks for this great work and project.

When i run following commands,

python ./DAGCN/train_advanced.py --model_name DAGCN_features --checkpoint_dir ./DAGCN/results/ --data_name CWRU --data_dir D:/Data/西储大学轴承数据中心网站 --transfer_task [3],[0] --last_batch True

I got following issues

ModuleNotFoundError: No module named 'torch_sparse'

Which version of torch do u use?

sunfeng90 avatar May 01 '22 10:05 sunfeng90