41 commits

Author	SHA1	Message	Date
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