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
bac11ae3
Commit
bac11ae3
authored
Feb 21, 2016
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix examples
parent
67fdbf26
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
12 deletions
+5
-12
example_alexnet.py
example_alexnet.py
+2
-6
example_cifar10.py
example_cifar10.py
+3
-6
No files found.
example_alexnet.py
View file @
bac11ae3
...
...
@@ -24,12 +24,8 @@ CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE
class
Model
(
ModelDesc
):
def
_get_input_vars
(
self
):
return
[
tf
.
placeholder
(
tf
.
float32
,
shape
=
(
None
,
227
,
227
,
3
),
name
=
'input'
),
tf
.
placeholder
(
tf
.
int32
,
shape
=
(
None
,),
name
=
'label'
)
]
return
[
InputVar
(
tf
.
float32
,
(
None
,
227
,
227
,
3
),
'input'
),
InputVar
(
tf
.
int32
,
(
None
,),
'label'
)
]
def
_get_cost
(
self
,
inputs
,
is_training
):
# img: 227x227x3
...
...
example_cifar10.py
View file @
bac11ae3
...
...
@@ -27,12 +27,9 @@ CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE
class
Model
(
ModelDesc
):
def
_get_input_vars
(
self
):
return
[
tf
.
placeholder
(
tf
.
float32
,
shape
=
[
None
,
30
,
30
,
3
],
name
=
'input'
),
tf
.
placeholder
(
tf
.
int32
,
shape
=
[
None
],
name
=
'label'
)
]
return
[
InputVar
(
tf
.
float32
,
[
None
,
30
,
30
,
3
],
'input'
),
InputVar
(
tf
.
int32
,
[
None
],
'label'
)
]
def
_get_cost
(
self
,
input_vars
,
is_training
):
image
,
label
=
input_vars
...
...
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