Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
3f14f3a7
Commit
3f14f3a7
authored
Apr 11, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix deprecate use of periodiccallback
parent
e49d4fd4
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
12 additions
and
6 deletions
+12
-6
docs/casestudies/colorize.md
docs/casestudies/colorize.md
+1
-1
examples/A3C-Gym/train-atari.py
examples/A3C-Gym/train-atari.py
+1
-1
examples/Char-RNN/char-rnn.py
examples/Char-RNN/char-rnn.py
+1
-2
tensorpack/RL/gymenv.py
tensorpack/RL/gymenv.py
+9
-2
No files found.
docs/casestudies/colorize.md
View file @
3f14f3a7
...
...
@@ -299,7 +299,7 @@ def get_config():
return
TrainConfig
(
dataflow
=
dataset
,
optimizer
=
tf
.
train
.
AdamOptimizer
(
lr
),
callbacks
=
[
Periodic
Callback
(
ModelSaver
(),
3
)],
callbacks
=
[
Periodic
Trigger
(
ModelSaver
(),
every_k_epochs
=
3
)],
model
=
Model
(),
steps_per_epoch
=
dataset
.
size
(),
max_epoch
=
100
,
...
...
examples/A3C-Gym/train-atari.py
View file @
3f14f3a7
...
...
@@ -220,7 +220,7 @@ def get_config():
[(
80
,
2
),
(
100
,
3
),
(
120
,
4
),
(
140
,
5
)]),
master
,
StartProcOrThread
(
master
),
Periodic
Callback
(
Evaluator
(
EVAL_EPISODE
,
[
'state'
],
[
'policy'
]),
2
),
Periodic
Trigger
(
Evaluator
(
EVAL_EPISODE
,
[
'state'
],
[
'policy'
]),
every_k_epochs
=
2
),
],
session_creator
=
sesscreate
.
NewSessionCreator
(
config
=
get_default_sess_config
(
0.5
)),
...
...
examples/Char-RNN/char-rnn.py
View file @
3f14f3a7
...
...
@@ -73,7 +73,7 @@ class Model(ModelDesc):
def
_build_graph
(
self
,
inputs
):
input
,
nextinput
=
inputs
cell
=
rnn
.
MultiRNNCell
([
rnn
.
BasicLSTM
Cell
(
num_units
=
param
.
rnn_size
)
cell
=
rnn
.
MultiRNNCell
([
rnn
.
LSTMBlock
Cell
(
num_units
=
param
.
rnn_size
)
for
_
in
range
(
param
.
num_rnn_layer
)])
def
get_v
(
n
):
...
...
@@ -91,7 +91,6 @@ class Model(ModelDesc):
input_list
=
tf
.
unstack
(
input_feature
,
axis
=
1
)
# seqlen x (Bxrnnsize)
# seqlen is 1 in inference. don't need loop_function
outputs
,
last_state
=
rnn
.
static_rnn
(
cell
,
input_list
,
initial
,
scope
=
'rnnlm'
)
self
.
last_state
=
tf
.
identity
(
last_state
,
'last_state'
)
...
...
tensorpack/RL/gymenv.py
View file @
3f14f3a7
...
...
@@ -84,7 +84,10 @@ except ImportError:
if
__name__
==
'__main__'
:
env
=
GymEnv
(
'Breakout-v0'
,
viz
=
0.1
)
import
gym_ple
,
cv2
# noqa
import
os
os
.
environ
[
"SDL_VIDEODRIVER"
]
=
"dummy"
env
=
GymEnv
(
'FlappyBird-v0'
,
viz
=
0.1
)
num
=
env
.
get_action_space
()
.
num_actions
()
from
..utils
import
get_rng
...
...
@@ -93,6 +96,10 @@ if __name__ == '__main__':
act
=
rng
.
choice
(
range
(
num
))
# print act
r
,
o
=
env
.
action
(
act
)
env
.
current_state
()
state
=
env
.
current_state
()
state
=
cv2
.
resize
(
state
[:
450
],
(
84
,
84
))
cv2
.
imshow
(
"aa"
,
state
)
cv2
.
waitKey
(
3
)
print
(
state
.
shape
)
if
r
!=
0
or
o
:
print
(
r
,
o
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment