Spaces:
Sleeping
Sleeping
fix: Remove leftover merge conflict marker in rl/training.py
Browse files- rl/training.py +1 -1
rl/training.py
CHANGED
|
@@ -240,10 +240,10 @@ def train_agent(
|
|
| 240 |
policy_cfg = policy_config or DEFAULT_POLICY_CONFIG
|
| 241 |
|
| 242 |
# Align agent hyperparameters with config
|
|
|
|
| 243 |
agent.discount = config.discount_factor
|
| 244 |
agent.epsilon = config.initial_epsilon
|
| 245 |
|
| 246 |
-
>>>>>>> origin/codex/modify-training-for-schedulingalgorithm-integration
|
| 247 |
training_stats = {
|
| 248 |
"episodes": [],
|
| 249 |
"total_rewards": [],
|
|
|
|
| 240 |
policy_cfg = policy_config or DEFAULT_POLICY_CONFIG
|
| 241 |
|
| 242 |
# Align agent hyperparameters with config
|
| 243 |
+
agent.learning_rate = config.learning_rate
|
| 244 |
agent.discount = config.discount_factor
|
| 245 |
agent.epsilon = config.initial_epsilon
|
| 246 |
|
|
|
|
| 247 |
training_stats = {
|
| 248 |
"episodes": [],
|
| 249 |
"total_rewards": [],
|