Skip to content

Commit 9899b79

Browse files
committed
Add action scaler for training continuous action-space dataset
1 parent 15c2ad2 commit 9899b79

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

minerva/tasks.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@ def train(algo_name,
5252
augmentation = DrQPipeline(augmentations, n_mean=n_mean)
5353
params['augmentation'] = augmentation
5454

55+
# add action scaler if continuous action-space
56+
if not dataset.is_action_discrete():
57+
params['action_scaler'] = 'min_max'
58+
5559
# train
5660
algo = create_algo(algo_name, dataset.is_action_discrete(), **params)
5761
algo.fit(train_data,

0 commit comments

Comments
 (0)