RoyAalekh commited on
Commit
5bbed6a
·
1 Parent(s): 8700d01

fix: Remove leftover merge conflict marker in rl/training.py

Browse files
Files changed (1) hide show
  1. 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": [],