Antonin RAFFIN
|
c264403816
|
Rename for consistency
+ add _predict to actors
+ improve sac actor code
|
2020-03-31 17:48:23 +02:00 |
|
Antonin RAFFIN
|
fdecd512db
|
Add save/load weights for policies and refactor action distributions
|
2020-03-31 16:29:13 +02:00 |
|
Antonin RAFFIN
|
72a88a8d92
|
Fix type hint for activation fn
|
2020-03-24 10:10:37 +01:00 |
|
Antonin RAFFIN
|
ba18258af6
|
Add proper preprocessing
|
2020-03-23 17:15:30 +01:00 |
|
Antonin RAFFIN
|
4b2092f55a
|
Remove base network
|
2020-03-23 15:31:14 +01:00 |
|
Antonin RAFFIN
|
b96a081e5f
|
Add check for image space
|
2020-03-20 11:20:57 +01:00 |
|
Antonin RAFFIN
|
57b37513b6
|
Refactor handling of obs and action space
+ remove duplicated code
|
2020-03-20 10:09:09 +01:00 |
|
Antonin RAFFIN
|
c3187604bc
|
Code cleanup: rename lr to lr_schedule + typing
|
2020-03-16 14:01:32 +01:00 |
|
Antonin RAFFIN
|
a67bb75438
|
Type td3 policies
|
2020-03-16 13:31:06 +01:00 |
|
Antonin Raffin
|
70e601c03c
|
Improve code and bump version
|
2020-03-12 15:34:35 +01:00 |
|
Antonin Raffin
|
18f38f8cf5
|
Reformat
|
2020-03-12 11:12:10 +01:00 |
|
Antonin Raffin
|
a2b1bf06d3
|
Add squash_output attribute to policy
|
2020-02-14 11:12:07 +01:00 |
|
Antonin Raffin
|
f1a4fa2d3f
|
Improve predict method
|
2020-02-12 15:25:05 +01:00 |
|
Antonin Raffin
|
b66003cfb3
|
Add callback support
|
2020-01-27 14:32:31 +01:00 |
|
Antonin Raffin
|
60d5f4463d
|
Add use_expln option for td3
|
2020-01-08 17:04:28 +01:00 |
|
Antonin Raffin
|
57708a628c
|
Add value function for SDE + TD3
|
2019-12-17 15:01:08 +01:00 |
|
Antonin Raffin
|
1ac1a7cad5
|
Reformat
|
2019-12-02 14:27:38 +01:00 |
|
Antonin Raffin
|
4e39a0627c
|
Refactor: enable sde net arch for TD3 and SAC
|
2019-12-02 14:06:17 +01:00 |
|
Antonin Raffin
|
5483e02d1a
|
Add SDE support for SAC
|
2019-11-26 15:26:12 +01:00 |
|
Antonin Raffin
|
d0003ee4ec
|
Enable kwargs for proba dist
|
2019-11-25 14:00:21 +01:00 |
|
Antonin Raffin
|
5bbb14188d
|
Unify A2C and TD3 SDE implementation
|
2019-11-25 13:19:33 +01:00 |
|
Antonin Raffin
|
c47be0086e
|
Add docstrings
|
2019-11-22 17:24:47 +01:00 |
|
Antonin Raffin
|
03d2ab10f8
|
Fix clipped action when adapting noise with TD3
|
2019-11-22 15:04:34 +01:00 |
|
Antonin Raffin
|
8aac10f3fa
|
Fix device
|
2019-11-13 14:38:18 +01:00 |
|
Antonin Raffin
|
da325a0ba7
|
Solve NaN issue and reduce number of parameters
|
2019-11-13 13:02:37 +01:00 |
|
Antonin Raffin
|
623e3cf4f9
|
Add SDE lr
|
2019-11-12 18:38:42 +01:00 |
|
Antonin Raffin
|
a08382faab
|
Add sde update for TD3
|
2019-11-12 18:37:13 +01:00 |
|
Antonin Raffin
|
f2a61949ae
|
Bump version and change default noise clipping
|
2019-11-08 14:59:42 +01:00 |
|
Antonin Raffin
|
715865a0fe
|
Add noise clipping
|
2019-11-08 13:17:38 +01:00 |
|
Antonin Raffin
|
db87e0d36a
|
Quick and dirty SDE version for TD3
|
2019-11-07 17:31:52 +01:00 |
|
Antonin Raffin
|
d67822718c
|
Add learning rate schedule
|
2019-10-28 16:47:13 +01:00 |
|
Antonin Raffin
|
b5656531d1
|
Enable logger for SAC/TD3 + refactor
|
2019-10-10 13:47:13 +02:00 |
|
Antonin Raffin
|
f4fe1362f0
|
Renaming
|
2019-09-24 14:53:03 +02:00 |
|
Antonin RAFFIN
|
2469ff3859
|
Reformat
|
2019-09-21 17:17:09 +02:00 |
|
Antonin Raffin
|
0e727a5f72
|
Full compat for VecEnv + bug fixes for cuda
|
2019-09-20 16:43:19 +02:00 |
|
Antonin Raffin
|
54dd7ea60d
|
Start PPO
|
2019-09-18 13:10:27 +02:00 |
|
Antonin Raffin
|
5e3a84d551
|
Refactor policies
|
2019-09-12 11:19:06 +02:00 |
|
Antonin Raffin
|
c3c87f8311
|
Add update styles for CEM-RL
|
2019-09-10 13:07:15 +02:00 |
|
Antonin Raffin
|
d22b66fc10
|
Fixes for CUDA support
|
2019-09-09 16:45:55 +02:00 |
|
Antonin Raffin
|
d4e2dc8a9c
|
Add CEM-RL
|
2019-09-06 14:01:10 +02:00 |
|
Antonin Raffin
|
904742714d
|
Fixes for python 2
|
2019-09-06 11:43:02 +02:00 |
|
Antonin Raffin
|
9cf289b997
|
Bug fixes + add evaluate script
|
2019-09-06 10:44:55 +02:00 |
|
Antonin Raffin
|
46d8d9725b
|
Init: TD3
|
2019-09-05 17:29:41 +02:00 |
|