Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
SparseConvNet
Commits
15fd91a0
Commit
15fd91a0
authored
Jan 10, 2018
by
Benjamin Thomas Graham
Browse files
remove tabs
parent
994d0b7e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
10 deletions
+10
-10
examples/hello-world.py
examples/hello-world.py
+10
-10
No files found.
examples/hello-world.py
View file @
15fd91a0
...
@@ -12,9 +12,9 @@ use_gpu = torch.cuda.is_available()
...
@@ -12,9 +12,9 @@ use_gpu = torch.cuda.is_available()
model
=
scn
.
Sequential
().
add
(
model
=
scn
.
Sequential
().
add
(
scn
.
SparseVggNet
(
2
,
1
,
scn
.
SparseVggNet
(
2
,
1
,
[[
'C'
,
8
],
[
'C'
,
8
],
[
'MP'
,
3
,
2
],
[[
'C'
,
8
],
[
'C'
,
8
],
[
'MP'
,
3
,
2
],
[
'C'
,
16
],
[
'C'
,
16
],
[
'MP'
,
3
,
2
],
[
'C'
,
16
],
[
'C'
,
16
],
[
'MP'
,
3
,
2
],
[
'C'
,
24
],
[
'C'
,
24
],
[
'MP'
,
3
,
2
]])
[
'C'
,
24
],
[
'C'
,
24
],
[
'MP'
,
3
,
2
]])
).
add
(
).
add
(
scn
.
ValidConvolution
(
2
,
24
,
32
,
3
,
False
)
scn
.
ValidConvolution
(
2
,
24
,
32
,
3
,
False
)
).
add
(
).
add
(
...
@@ -40,10 +40,10 @@ msg = [
...
@@ -40,10 +40,10 @@ msg = [
input
.
addSample
()
input
.
addSample
()
for
y
,
line
in
enumerate
(
msg
):
for
y
,
line
in
enumerate
(
msg
):
for
x
,
c
in
enumerate
(
line
):
for
x
,
c
in
enumerate
(
line
):
if
c
==
'X'
:
if
c
==
'X'
:
location
=
torch
.
LongTensor
([
y
,
x
])
location
=
torch
.
LongTensor
([
y
,
x
])
featureVector
=
torch
.
FloatTensor
([
1
])
featureVector
=
torch
.
FloatTensor
([
1
])
input
.
setLocation
(
location
,
featureVector
,
0
)
input
.
setLocation
(
location
,
featureVector
,
0
)
#Add a sample using setLocations
#Add a sample using setLocations
input
.
addSample
()
input
.
addSample
()
...
@@ -51,9 +51,9 @@ locations = []
...
@@ -51,9 +51,9 @@ locations = []
features
=
[]
features
=
[]
for
y
,
line
in
enumerate
(
msg
):
for
y
,
line
in
enumerate
(
msg
):
for
x
,
c
in
enumerate
(
line
):
for
x
,
c
in
enumerate
(
line
):
if
c
==
'X'
:
if
c
==
'X'
:
locations
.
append
([
y
,
x
])
locations
.
append
([
y
,
x
])
features
.
append
([
1
])
features
.
append
([
1
])
locations
=
torch
.
LongTensor
(
locations
)
locations
=
torch
.
LongTensor
(
locations
)
features
=
torch
.
FloatTensor
(
features
)
features
=
torch
.
FloatTensor
(
features
)
input
.
setLocations
(
locations
,
features
,
0
)
input
.
setLocations
(
locations
,
features
,
0
)
...
...
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