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
15c0e160
Commit
15c0e160
authored
Apr 26, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Handle "long" type in Python 2 (#737)
parent
a2c36b3d
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
3 deletions
+4
-3
examples/ResNet/README.md
examples/ResNet/README.md
+2
-2
tensorpack/dataflow/common.py
tensorpack/dataflow/common.py
+2
-1
No files found.
examples/ResNet/README.md
View file @
15c0e160
...
...
@@ -7,8 +7,8 @@ __Training__ code of three variants of ResNet on ImageNet:
*
[
Pre-activation ResNet
](
https://arxiv.org/abs/1603.05027
)
*
[
Squeeze-and-Excitation ResNet
](
https://arxiv.org/abs/1709.01507
)
The training
mostly follows the setup in
[
fb.resnet.torch
](
https://github.com/facebook/fb.resnet.torch
)
and gets
similar performance (with much fewer lines of code)
.
The training
follows the __exact__ recipe used by the
[
Training ImageNet in 1 Hour paper
](
https://arxiv.org/abs/1706.02677
)
and gets
the same performance
.
Models can be
[
downloaded here
](
http://models.tensorpack.com/ResNet/
)
.
| Model | Top 5 Error | Top 1 Error |
...
...
tensorpack/dataflow/common.py
View file @
15c0e160
...
...
@@ -3,6 +3,7 @@
from
__future__
import
division
import
six
import
numpy
as
np
from
copy
import
copy
import
pprint
...
...
@@ -131,7 +132,7 @@ class BatchData(ProxyDataFlow):
[
x
[
k
]
for
x
in
data_holder
])
else
:
dt
=
data_holder
[
0
][
k
]
if
type
(
dt
)
in
[
int
,
bool
]:
if
type
(
dt
)
in
list
(
six
.
integer_types
)
+
[
bool
]:
tp
=
'int32'
elif
type
(
dt
)
==
float
:
tp
=
'float32'
...
...
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