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
f1fc7337
Commit
f1fc7337
authored
Jun 13, 2016
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix debugging bug in expreplay
parent
e5a48033
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
2 additions
and
5 deletions
+2
-5
tensorpack/RL/expreplay.py
tensorpack/RL/expreplay.py
+1
-3
tensorpack/models/batch_norm.py
tensorpack/models/batch_norm.py
+0
-1
tensorpack/train/trainer.py
tensorpack/train/trainer.py
+1
-1
No files found.
tensorpack/RL/expreplay.py
View file @
f1fc7337
...
@@ -72,9 +72,7 @@ class ExpReplay(DataFlow, Callback):
...
@@ -72,9 +72,7 @@ class ExpReplay(DataFlow, Callback):
with
tqdm
(
total
=
self
.
init_memory_size
)
as
pbar
:
with
tqdm
(
total
=
self
.
init_memory_size
)
as
pbar
:
while
len
(
self
.
mem
)
<
self
.
init_memory_size
:
while
len
(
self
.
mem
)
<
self
.
init_memory_size
:
from
copy
import
deepcopy
self
.
_populate_exp
()
self
.
mem
.
append
(
deepcopy
(
self
.
mem
[
0
]))
#self._populate_exp()
pbar
.
update
()
pbar
.
update
()
self
.
_init_memory_flag
.
set
()
self
.
_init_memory_flag
.
set
()
...
...
tensorpack/models/batch_norm.py
View file @
f1fc7337
...
@@ -64,7 +64,6 @@ def BatchNorm(x, use_local_stat=True, decay=0.9, epsilon=1e-5):
...
@@ -64,7 +64,6 @@ def BatchNorm(x, use_local_stat=True, decay=0.9, epsilon=1e-5):
ema_apply_op
=
ema
.
apply
([
batch_mean
,
batch_var
])
ema_apply_op
=
ema
.
apply
([
batch_mean
,
batch_var
])
ema_mean
,
ema_var
=
ema
.
average
(
batch_mean
),
ema
.
average
(
batch_var
)
ema_mean
,
ema_var
=
ema
.
average
(
batch_mean
),
ema
.
average
(
batch_var
)
G
=
tf
.
get_default_graph
()
G
=
tf
.
get_default_graph
()
try
:
try
:
mean_name
=
re
.
sub
(
'towerp[0-9]+/'
,
''
,
ema_mean
.
name
)
mean_name
=
re
.
sub
(
'towerp[0-9]+/'
,
''
,
ema_mean
.
name
)
...
...
tensorpack/train/trainer.py
View file @
f1fc7337
...
@@ -90,9 +90,9 @@ class EnqueueThread(threading.Thread):
...
@@ -90,9 +90,9 @@ class EnqueueThread(threading.Thread):
pass
pass
except
Exception
:
except
Exception
:
logger
.
exception
(
"Exception in EnqueueThread:"
)
logger
.
exception
(
"Exception in EnqueueThread:"
)
finally
:
self
.
sess
.
run
(
self
.
close_op
)
self
.
sess
.
run
(
self
.
close_op
)
self
.
coord
.
request_stop
()
self
.
coord
.
request_stop
()
finally
:
logger
.
info
(
"Enqueue Thread Exited."
)
logger
.
info
(
"Enqueue Thread Exited."
)
class
QueueInputTrainer
(
Trainer
):
class
QueueInputTrainer
(
Trainer
):
...
...
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