mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-10 10:21:59 +00:00
Update BaseReinforcementLearningModel.py
Add support for DDPG and TD3.
This commit is contained in:
parent
dc5766fb10
commit
c83dd2d806
|
@ -32,7 +32,7 @@ logger = logging.getLogger(__name__)
|
|||
|
||||
torch.multiprocessing.set_sharing_strategy("file_system")
|
||||
|
||||
SB3_MODELS = ["PPO", "A2C", "DQN"]
|
||||
SB3_MODELS = ["PPO", "A2C", "DQN", "DDPG", "TD3"]
|
||||
SB3_CONTRIB_MODELS = ["TRPO", "ARS", "RecurrentPPO", "MaskablePPO", "QRDQN"]
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user