File size: 106 Bytes
f238a34
 
1
2
3
from stable_baselines3.dqn.dqn import DQN
from stable_baselines3.dqn.policies import CnnPolicy, MlpPolicy