Try to fix type error
Browse files
app.py
CHANGED
@@ -50,9 +50,6 @@ if __name__ == "__main__":
|
|
50 |
gr.Warning(
|
51 |
"We're currently in the process of adding support for Mujoco environments, so the application may encounter crashes during this phase. We encourage contributors to join us in the repository https://github.com/riiswa/kanrl to assist in the development and support of other environments. Your contributions are invaluable in ensuring a robust and comprehensive framework."
|
52 |
)
|
53 |
-
torch.set_default_dtype(torch.float64)
|
54 |
-
else:
|
55 |
-
torch.set_default_dtype(torch.float32)
|
56 |
agent = "ppo"
|
57 |
if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
|
58 |
agent ="trpo"
|
|
|
50 |
gr.Warning(
|
51 |
"We're currently in the process of adding support for Mujoco environments, so the application may encounter crashes during this phase. We encourage contributors to join us in the repository https://github.com/riiswa/kanrl to assist in the development and support of other environments. Your contributions are invaluable in ensuring a robust and comprehensive framework."
|
52 |
)
|
|
|
|
|
|
|
53 |
agent = "ppo"
|
54 |
if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
|
55 |
agent ="trpo"
|