improve pretrain

2026-07-15 16:56:54 +08:00 · 2021-07-28 10:07:35 +08:00
parent 5b7b48e376
commit a7c41b6969
4 changed files with 9 additions and 5 deletions
--- a/examples/benchmarks/TRA/workflow_config_tra_Alpha158.yaml
+++ b/examples/benchmarks/TRA/workflow_config_tra_Alpha158.yaml
@@ -73,6 +73,7 @@ task:
    kwargs:
      tra_config: *tra_config
      model_config: *model_config
+      model_type: RNN
      lr: 1e-3
      n_epochs: 100
      max_steps_per_epoch: 100
@@ -86,7 +87,7 @@ task:
      memory_mode: *memory_mode
      eval_train: False
      eval_test: True
-      pretrain: False
+      pretrain: True
      init_state:
      freeze_model: False
      freeze_predictors: False