Блог

Я тренирую простую нейронную сеть на встраивании слов. Я пытаюсь сохранять сюжет вложений каждые 1000 эпох:

 if epo00 == 0:
    print(f'Epoch {epo}')

    #save plots
    W = net.fc1.weight
    W = W.detach().numpy()

    svd = decomposition.TruncatedSVD(n_components=2)
    W1_dec = svd.fit_transform(W.T)
    x = W1_dec[:,0]
    y = W1_dec[:,1]
    plot = sns.scatterplot(x, y)

    for i in range(0,W1_dec.shape[0]):
        plot.text(x[i], y[i] 2e-2, list(vocabulary.keys())[i], horizontalalignment='center', size='small', color='black', weight='semibold')
    plt.savefig(f'images/epoch{epo}.png')
 

Сначала обучение и сюжеты будут отличными, но через некоторое время они значительно замедлятся. Я уже сталкивался с этой проблемой раньше, когда каждый раз использовался весь вычислительный график, но я не вижу, где я мог бы сделать то же самое здесь.

Я попытался удалить часть построения (в разделе «#сохранить графики»), и все работает просто отлично и не замедляется.

Обновить

Похоже, что последняя строка-это та, которая замедляет процесс: plt.savefig(f'images/epoch{epo}.png')

Не знаю, как это исправить