30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: UTF-8 -*-
|
|
###########################################################################
|
|
# Copyright © 1998 - 2026 Tencent. All Rights Reserved.
|
|
###########################################################################
|
|
"""
|
|
Author: Tencent AI Arena Authors
|
|
"""
|
|
|
|
from kaiwudrl.common.utils.train_test_utils import run_train_test
|
|
|
|
# To run the train_test, you must modify the algorithm name here. It must be one of algorithm_name_list.
|
|
# Simply modify the value of the algorithm_name variable.
|
|
# 运行train_test前必须修改这里的算法名字, 必须是 algorithm_name_list 里的一个, 修改algorithm_name的值即可
|
|
algorithm_name_list = ["ppo", "diy"]
|
|
algorithm_name = "ppo"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_train_test(
|
|
algorithm_name=algorithm_name,
|
|
algorithm_name_list=algorithm_name_list,
|
|
env_vars={
|
|
"replay_buffer_capacity": "8",
|
|
"preload_ratio": "0.1",
|
|
"train_batch_size": "2",
|
|
"dump_model_freq": "100",
|
|
},
|
|
)
|