mirror of
https://github.com/wassname/ETSformer.git
synced 2026-06-27 16:28:38 +08:00
minor refactor
This commit is contained in:
+3
-3
@@ -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)
|
||||
|
||||
@@ -0,0 +1 @@
|
||||
from .etsformer.model import ETSformer
|
||||
|
||||
Reference in New Issue
Block a user