训练的时候:
python scripts/train.py --delim=“tab” --noise_dim=0
运行已有模型的时候:
PYTHONPATH=. python scripts/evaluate_model.py --model_path models/sgan-model
但是cuda版本不对
需要将10.1换成8.0
首先在cuda目录下打开终端,运行
gedit ~/.bashrc
后把版本给改一下
之后sudo rm -rf cuda,关闭,重新打开输入
nvcc -V 发现已经改变。
在尝试运行的过程中,报了以下错误
- valueerror:could not convert string to float
修改方法:
lines = f.readlines()
for line in lines:
line = line.strip("\n")
line = line.split(" ")
line = [float(x) for x in line]
参考链接:https://blog.csdn.net/weixin_35436966/article/details/105941671
2. runtimeerror: view size is not compatible with input tensor’s size and stride:
解决方法:加上.contiguous()
out = out.contiguous().view(out.size()[0], -1)
参考链接:https://blog.csdn.net/tiao_god/article/details/108189879