Почему мое ядро ​​умирает и перезагружается во время обучения моей нейронной сети с помощью PyTorch? - PullRequest
Новая
       29

Почему мое ядро ​​умирает и перезагружается во время обучения моей нейронной сети с помощью PyTorch?

0 голосов
/ 15 марта 2019

Я новичок в Python и PyTorch, и мне нужно реализовать нейронную сеть для распознавания изображений.

Все работает нормально, за исключением того, что мое ядро ​​умирает и автоматически перезапускается на 78-й итерации.На каждой итерации мы обучаем сеть из 64 изображений (в полном наборе 5000 изображений).

Я использую Macbook Pro (10.14), и я установил Pytorch с помощью pip.Это первый раз, когда я сталкиваюсь с этой проблемой.

Я немного копаюсь в интернете, и это может быть проблема с памятью?

Спасибо за чтение! 

 epochs = 10
 steps = 0
 print_every = 20
 running_loss = 0 
 for e in range(epochs):
 start = time.time()

 for i, (images, labels) in enumerate(iter(valloader)):
    steps += 1

    images.resize_(64, 3072)
    optimizer.zero_grad()
    output = net.forward(images)
    loss = loss_func(output, labels)
    loss.backward()
    optimizer.step()

    running_loss += loss.item()
...