Initial robot vacuum code

train_test.py

@@ -0,0 +1,29 @@
+#!/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": "10",
+            "preload_ratio": "0.2",
+            "train_batch_size": "2",
+            "dump_model_freq": "1",
+        },
+    )