CUDA未能启动运行时seq2seq模型
1视图(30天)
显示旧的评论
当我训练序列序列模型。这个错误发生,
警告:在CUDA执行期间发生意外错误。CUDA的错误是:CUDA_ERROR_LAUNCH_FAILED
在那之后,我再也不能使用GPU运行函数gpuArray ()。没有问题,当我训练任何图像分类模型后重新启动Matlab或之前运行该模型。
模型:#样品1250年的每一部分
层= […
sequenceInputLayer (40)
bilstmLayer (200“OutputMode”,“序列”)
dropoutLayer (0.5)
bilstmLayer (150“OutputMode”,“序列”)
dropoutLayer (0.5)
bilstmLayer (100“OutputMode”,“序列”)
dropoutLayer (0.5)
fullyConnectedLayer (4)
softmaxLayer
classificationLayer];
选择= trainingOptions (“亚当”,…
“MaxEpochs”,20岁,…
“MiniBatchSize”32岁的…
“InitialLearnRate”,0.01,…
“LearnRateDropPeriod”5,…
“LearnRateSchedule”,“分段”,…
“ValidationData”,{fsstValidationData (: 1), labelValidationData},…
“ValidationFrequency”5,…
“ValidationPatience”正,…
“GradientThreshold”,1…
“阴谋”,“训练进步”,…
“洗牌”,“every-epoch”,…
“详细”0,…
“DispatchInBackground”,真正的);
GPU: RTX 2060超级8 GB;