Skip to content

Commit 502d85f

Browse files
committed
Update readme
1 parent 14a7546 commit 502d85f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ dataset, env = get_atari('breakout-expert-v0')
114114
train_episodes, test_episodes = train_test_split(dataset, test_size=0.2)
115115

116116
# setup algorithm
117-
cql = DiscreteCQL(n_frames=4, q_func_type='qr', scaler='pixel', use_gpu=True)
117+
cql = DiscreteCQL(n_frames=4, q_func_factory='qr', scaler='pixel', use_gpu=True)
118118

119119
# start training
120120
cql.fit(train_episodes,
@@ -145,7 +145,7 @@ dataset, env = get_pybullet('hopper-bullet-mixed-v0')
145145
train_episodes, test_episodes = train_test_split(dataset, test_size=0.2)
146146

147147
# setup algorithm
148-
cql = CQL(q_func_type='qr', use_gpu=True)
148+
cql = CQL(q_func_factory='qr', use_gpu=True)
149149

150150
# start training
151151
cql.fit(train_episodes,

0 commit comments

Comments
 (0)