小编xia*_*hen的帖子

如何修复这个奇怪的错误:“运行时错误:CUDA 错误:内存不足”

我运行了一个关于深度学习网络的代码,首先我训练了网络,它运行良好,但是运行到验证网络时出现此错误。

我有五个 epoch,每个 epoch 都有一个训练和验证的过程。我在第一个纪元验证时遇到了错误。所以我没有运行验证代码,我发现代码可以运行到第二个纪元并且没有错误。

我的代码:

for epoch in range(10,15): # epoch: 10~15
    if(options["training"]["train"]):
        trainer.epoch(model, epoch)

    if(options["validation"]["validate"]):
    #if(epoch == 14):
        validator.epoch(model)
Run Code Online (Sandbox Code Playgroud)

在此处输入图片说明 在此处输入图片说明

我觉得验证代码可能有一些错误。但我找不到那个。

python pycharm pytorch

32
推荐指数
6
解决办法
12万
查看次数

标签 统计

pycharm ×1

python ×1

pytorch ×1