Adding preprocessing function

2023-04-12 17:16:47 +02:00 · 2023-04-12 17:16:47 +02:00 · 1f4667a08d
commit 1f4667a08d
parent 7d7387bd5d
1 changed files with 5 additions and 1 deletions
--- a/DPI/utils.py
+++ b/DPI/utils.py
@ -200,6 +200,10 @@ def make_env(args):
        )
    return env

+def preprocess_obs(obs):
+    obs = obs/255.0 - 0.5
+    return obs
+
 def soft_update_params(net, target_net, tau):
    for param, target_param in zip(net.parameters(), target_net.parameters()):
        target_param.data.copy_(