We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 14a7546 commit 502d85fCopy full SHA for 502d85f
README.md
@@ -114,7 +114,7 @@ dataset, env = get_atari('breakout-expert-v0')
114
train_episodes, test_episodes = train_test_split(dataset, test_size=0.2)
115
116
# setup algorithm
117
-cql = DiscreteCQL(n_frames=4, q_func_type='qr', scaler='pixel', use_gpu=True)
+cql = DiscreteCQL(n_frames=4, q_func_factory='qr', scaler='pixel', use_gpu=True)
118
119
# start training
120
cql.fit(train_episodes,
@@ -145,7 +145,7 @@ dataset, env = get_pybullet('hopper-bullet-mixed-v0')
145
146
147
148
-cql = CQL(q_func_type='qr', use_gpu=True)
+cql = CQL(q_func_factory='qr', use_gpu=True)
149
150
151
0 commit comments