第一次使用DGLke,用的是TransE_I2模型,不太懂它的原理,跟着官方文档修改了参数以后,出现了下面的问题,看不明白是发生了什么,也不知道从哪开始解决
使用该命令:DGLBACKEND=pytorch dglke_train --dataset TE --data_path D:/personal path --data_files train.tsv valid.tsv test.tsv --format 'raw_udd_hrt' --model_name TransE_l2 --batch_size 2048 --neg_sample_size 256 --hidden_dim 400 --gamma 12.0 --lr 0.1 --max_step 100000 --log_interval 1000 --batch_size_eval 16 -adv --regularization_coef 1.00E-07 --test --num_thread 1 --gpu 0 --neg_sample_size_eval 10000 --async_update
D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\base.py:45: DGLWarning: Recommend creating graphs by dgl.graph(data)
instead of dgl.DGLGraph(data)
. return warnings.warn(message, category=category, stacklevel=1)
D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\base.py:45: DGLWarning: Keyword arguments ['readonly', 'multigraph', 'sort_csr'] are deprecated in v0.5, and can be safe
ly removed in all cases.
return warnings.warn(message, category=category, stacklevel=1)
Traceback (most recent call last):
File "D:\Software\anaconda3\envs\pytorch\lib\runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dglke\dataloader\sampler.py", line 379, in create_sampler
return EdgeSampler(self.g,
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\contrib\sampling\sampler.py", line 662, in init
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\utils\internal.py", line 271, in toindex
return data if isinstance(data, Index) else Index(data, dtype)
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\utils\internal.py", line 26, in init
self._initialize_data(data)
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\utils\internal.py", line 33, in _initialize_data
self._dispatch(data)
File "D:\Software\anaconda3\envs\pytorch\lib\site-packages\dgl\utils\internal.py", line 58, in _dispatch
raise InconsistentDtypeException('Index data specified as %s, but got: %s' %
dgl.utils.internal.InconsistentDtypeException: DGL now requires the input tensor to have the same dtype as the graph index's dtype(which you can get by g.idype). Inde
x data specified as int64, but got: int32
我想知道一下这样的报错是什么原因,怎么解决
你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答
本次提问扣除的有问必答次数,将会以问答VIP体验卡(1次有问必答机会、商城购买实体图书享受95折优惠)的形式为您补发到账户。
因为有问必答VIP体验卡有效期仅有1天,您在需要使用的时候【私信】联系我,我会为您补发。
我仔细看了一下dglke的官方文档,发现它不支持windows系统,当我改用linux系统时,就没有了上述问题
您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!