32 commits

Author	SHA1	Message	Date
Antonin RAFFIN	7ae54206ce	Reformat and code cleanup	2020-04-23 15:18:21 +02:00
Antonin RAFFIN	71df3c7409	Add docstrings and missing types	2020-04-23 14:56:05 +02:00
Antonin RAFFIN	f38ddcb278	Allow any number of channels	2020-04-22 16:11:23 +02:00
Antonin RAFFIN	041f2bc59a	Cleanup, bug fixes + more tests	2020-04-22 13:14:22 +02:00
Antonin RAFFIN	73fb8d1c63	Add CNN support for TD3	2020-04-22 11:05:46 +02:00
Antonin RAFFIN	b289aca5fe	Bug fixes at loading and predict time	2020-04-21 21:06:07 +02:00
Antonin RAFFIN	8aac9e819d	Add VecTransposeImage and fix for SAC	2020-04-21 20:41:58 +02:00
Antonin RAFFIN	93c2a01f91	Start CNN support (failing for SAC)	2020-04-21 16:22:46 +02:00
Antonin RAFFIN	6bd2d87f33	Improve doc	2020-04-20 16:21:47 +02:00
Antonin RAFFIN	f347474e6a	Independent save/load for policies	2020-04-20 15:59:44 +02:00
Antonin RAFFIN	a4df08cd28	Complete save/load for TD3 policy	2020-04-20 15:43:23 +02:00
Antonin RAFFIN	17f9246257	Add get_device util and fix squash_output	2020-04-20 15:43:11 +02:00
Antonin RAFFIN	aa1026ee87	Added `optimizer and optimizer_kwargs to policy_kwargs`	2020-04-17 15:13:45 +02:00
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