We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 15c2ad2 commit 9899b79Copy full SHA for 9899b79
minerva/tasks.py
@@ -52,6 +52,10 @@ def train(algo_name,
52
augmentation = DrQPipeline(augmentations, n_mean=n_mean)
53
params['augmentation'] = augmentation
54
55
+ # add action scaler if continuous action-space
56
+ if not dataset.is_action_discrete():
57
+ params['action_scaler'] = 'min_max'
58
+
59
# train
60
algo = create_algo(algo_name, dataset.is_action_discrete(), **params)
61
algo.fit(train_data,
0 commit comments