22 lines
480 B
Python
22 lines
480 B
Python
def atari():
|
|
return dict(
|
|
network='conv_only',
|
|
lr=1e-4,
|
|
buffer_size=10000,
|
|
exploration_fraction=0.1,
|
|
exploration_final_eps=0.01,
|
|
train_freq=4,
|
|
learning_starts=10000,
|
|
target_network_update_freq=1000,
|
|
gamma=0.99,
|
|
prioritized_replay=True,
|
|
prioritized_replay_alpha=0.6,
|
|
checkpoint_freq=10000,
|
|
checkpoint_path=None,
|
|
dueling=True
|
|
)
|
|
|
|
def retro():
|
|
return atari()
|
|
|