Fork me on Gitee

随笔分类 -  PyTorch

摘要:# 测试 Embedding import torch import gensim import torch.nn as nn wvmodel = gensim.models.KeyedVectors.load_word2vec_format("./data/vector.txt",binary=F 阅读全文
posted @ 2023-03-24 14:45 ---dgw博客 阅读(88) 评论(0) 推荐(0) 编辑
摘要:解决办法:window登录管理员用户,打开NSIGHT ,设置option的Microsoft display driver 的WDDM TDR enable为False即可,因为该选项导致kernel运行时间超过指定值会中断 阅读全文
posted @ 2020-05-11 11:39 ---dgw博客 阅读(6527) 评论(0) 推荐(0) 编辑
摘要:出现问题:我们访问这个地址:页面最下面找到Windows支持:Windows支持Windows支持是实验性的,建议使用Linux。 pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .如果您能够从系统上的源代码构建Pytorch,则可能会起作用。 pip install -v... 阅读全文
posted @ 2020-05-11 00:49 ---dgw博客 阅读(1472) 评论(0) 推荐(0) 编辑
摘要:具体错误日志如下:The resulting error log is as followsTraceback (most recent call last): File "train.py", line 441, in train() # train normally File "train.py", line 324, in train dataloader=testloader) File... 阅读全文
posted @ 2020-01-25 01:38 ---dgw博客 阅读(3757) 评论(0) 推荐(0) 编辑
摘要:$ python train.py --data data/coco.data --cfg cfg/yolov3.cfgNamespace(accumulate=2, adam=False, arc='default', batch_size=32, bucket='', cache_images=False, cfg='cfg/yolov3.cfg', data='data/coco.data'... 阅读全文
posted @ 2020-01-21 21:03 ---dgw博客 阅读(937) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示