19 commits

Author	SHA1	Message	Date
Antonin RAFFIN	6e470d0f72	Fix: don't change the device when saving	2020-03-31 18:38:37 +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	35d0d2b320	More typing	2020-03-10 18:09:45 +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	88f07bafb6	Convert format to f-strings	2020-01-22 16:39:25 +01:00
Antonin Raffin	4e39a0627c	Refactor: enable sde net arch for TD3 and SAC	2019-12-02 14:06:17 +01:00
Antonin Raffin	7a6a500398	Allow to use states directly as features for sde	2019-12-02 11:48:16 +01:00
Antonin Raffin	b84e5e9e27	Move flexible mlp to common	2019-11-22 13:06:41 +01:00
Antonin Raffin	b4dc9d4e4d	Add doc	2019-09-26 11:46:40 +02:00
Antonin Raffin	322399e8fe	Update collect rollout	2019-09-25 13:20:06 +02:00
Antonin RAFFIN	6bb7e183d2	Running PPO (not working yet)	2019-09-18 15:35:17 +02:00
Antonin Raffin	54dd7ea60d	Start PPO	2019-09-18 13:10:27 +02:00
Antonin Raffin	d4e2dc8a9c	Add CEM-RL	2019-09-06 14:01:10 +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