Python (Pytorch) - циклы и логирование

Question

Я очень новичок в Python и мне нужна помощь по циклу ANN с различным числом эпох.

Код начинается с

import sys
from tqdm import tqdm
import numpy as np

import torch
import torch.nn.functional as F

try:

и части эпохи

for t in range(2000):
prediction = net(x)

Могу ли я спросить, что мне добавить (до и после текущего кода), чтобы зациклить программу 3 раза, затем добавить 1000 эпох и запускать снова, пока эпоха не достигнет 10000?

Большое спасибо.

Кроме того, я хотел бы знать после цикла, как вести журнал результатов?

loss_func = torch.nn.MSELoss() 

Answer 1

Обновите цикл for как:

import sys

# To log your outputs in a separate file "myErrorLog.txt"
sys.stdout = open("myErrorLog.txt", "w")
for t in range(2000, 10000, 1000):
    print ("iteration {}".format(t))
    # your work here

# end of your code
sys.stdout.close()