Comments (6)
train_dataloader = TrainDataLoader(
in_path = "./benchmarks/MetaQA/TransE/",
nbatches = 100,
threads = 8,
sampling_mode = "normal",
bern_flag = 1,
filter_flag = 1,
neg_ent = 25,
neg_rel = 0)
add "/" at the end of in_path line, and then solved the Segmentation fault problem.
from openke.
抱歉再来打扰一下,我执行train_rotate_WN18RR_adv.py(openke自带的rotate例子)的时候会出现:Finish initializing...
0%| | 0/1500 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/mnt/d/learning/kgqa/KGQA-master/OpenKE/train_rotate.py", line 47, in
trainer.run()
File "/mnt/d/learning/kgqa/KGQA-master/OpenKE/openke/config/Trainer.py", line 94, in run
loss = self.train_one_step(data)
File "/mnt/d/learning/kgqa/KGQA-master/OpenKE/openke/config/Trainer.py", line 53, in train_one_step
loss.backward()
File "/home/stone/.local/lib/python3.10/site-packages/torch/_tensor.py", line 487, in backward
torch.autograd.backward(
File "/home/stone/.local/lib/python3.10/site-packages/torch/autograd/init.py", line 200, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
RuntimeError: CUDA error: unknown error
Compile with TORCH_USE_CUDA_DSA
to enable device-side assertions.
这样的情况,想问一下这个是由于cuda和pytorch版本对不上的问题吗?我用的是wsl+ubuntu,但是运行其他模型的训练例子代码都没什么问题,想请教一下有没有什么解决的思路。
from openke.
我也有segmentation fault的错误,请问你是怎么解决的?存储格式也是按照examples里数据集改的
from openke.
Related Issues (20)
- Normal and Cross Sampling
- Question about loss? HOT 1
- segmentation fault when using custom dataset HOT 2
- cannot reshape tensor of 0 elements into shape [-1, 0] because the unspecified dimension size -1 can be any value and is ambiguous
- Traceback (most recent call last): HOT 1
- Segmentation fault (core dumped) HOT 1
- RuntimeError: unique_by_key: failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered HOT 1
- 关于Calloc内存问题 HOT 3
- 是否可以仅使用train.txt训练得到embedding,而不用entity2id.txt以及relation2id.txt这两个文件? HOT 2
- Process finished with exit code 139
- Issues with Config.cy HOT 1
- Experiments results 每个数据集每个模型的超参数?
- corrupt_head函数问题
- malloc(): invalid size (unsorted) HOT 1
- 预测结果和指标计算问题
- 自定义数据集,loss不收敛
- Train tripes - Query on Duplicate elimination
- No exception! Everytime a small error would crush, really hard to debug
- Training on Wikidata (huge dataset) using OpenKE
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from openke.