[Example]fix bug of transformer4sr (PaddlePaddle#1017)

* [Example]fix bug of transformer4sr * Update examples/transformer4sr/conf/transformer4sr.yaml --------- Co-authored-by: HydrogenSulfate <[email protected]>
PolaKuma · Nov 14, 2024 · 615686a · 615686a
1 parent 474bf7b
commit 615686a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/examples/transformer4sr/conf/transformer4sr.yaml b/examples/transformer4sr/conf/transformer4sr.yaml
@@ -47,7 +47,7 @@ DATA_GENERATE:
   num_zfill: 8
 DATA:
   data_path: "./data_generated/" # ${DATA_GENERATE.data_path}
-  data_path_srsd: ["./srsd-feynman_easy/", "./srsd-feynman_medium/", "./srsd-feynman_hard/"]
+  data_path_srsd: ["./srsd-feynman_easy/"]
   ratio: [0.8,0.1,0.1]
   sampling_times: ${DATA_GENERATE.sampling_times}
   seq_length_max: 30  # ${DATA_GENERATE.seq_length_max}

diff --git a/examples/transformer4sr/transformer4sr.py b/examples/transformer4sr/transformer4sr.py
@@ -150,8 +150,7 @@ def evaluate(cfg: DictConfig):
         vocab_size=vocab_size,
         seq_length=data_funcs.seq_length_max,
     )
-    param_dict = paddle.load(cfg.EVAL.pretrained_model_path)
-    model.set_state_dict(param_dict)
+    ppsci.utils.save_load.load_pretrain(model, path=cfg.EVAL.pretrained_model_path)
     model.eval()
 
     # evaluate

diff --git a/ppsci/arch/transformer.py b/ppsci/arch/transformer.py
@@ -331,7 +331,6 @@ def __init__(
         self.act = act
         self.dropout = dropout
 
-        super().__init__()
         self.encoder = Encoder(
             num_layers_enc, num_var_max, d_model, heads, act="relu", dropout=dropout
         )