How to save trained model to a file and apply the model? #64
Answered
by
PaParaZz1
zhangjingwei713
asked this question in
Q&A
-
I tested the test_smac_env.py and although it can run, the trained model is not saved. how can I save the model to a file, and how can I use the trained model like sl_model.pth for DI-star? Thanks! |
Beta Was this translation helpful? Give feedback.
Answered by
PaParaZz1
Sep 23, 2021
Replies: 1 comment 3 replies
-
|
Beta Was this translation helpful? Give feedback.
3 replies
Answer selected by
PaParaZz1
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
test_smac_env.py
(link) is just a unittest file for smac environment without any training code.ding -m serial -c dizoo/smac/config/smac_3s5z_qmix_config.py -s 0
in your terminal, and model will be saved automatically indefault_experiment/ckpt
directory when better final reward is acquired.