minor refactor

This commit is contained in:
gorold
2022-09-05 22:42:24 +08:00
parent 544943e810
commit e584961249
8 changed files with 4 additions and 3 deletions
+3 -3
View File
@@ -1,4 +1,4 @@
from models.ETSformer.model import ETSformer
from models import ETSformer
from data_provider.data_factory import data_provider
from exp.exp_basic import Exp_Basic
from utils.tools import EarlyStopping, adjust_learning_rate
@@ -210,8 +210,8 @@ class Exp_Main(Exp_Basic):
outputs = outputs.detach().cpu().numpy()
batch_y = batch_y.detach().cpu().numpy()
pred = outputs # outputs.detach().cpu().numpy() # .squeeze()
true = batch_y # batch_y.detach().cpu().numpy() # .squeeze()
pred = outputs
true = batch_y
preds.append(pred)
trues.append(true)
+1
View File
@@ -0,0 +1 @@
from .etsformer.model import ETSformer