Изменения

for epoch in range(EPOCH):

losses = []

if RESCHEDULED == False and epoch == EPOCH // 2:

LR *= 0.1