Commit 15fd91a0 authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

remove tabs

parent 994d0b7e
...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment