Skip to content

Commit adcb718

Browse files
authored
Fixes for V0.2 (pytorch#122)
* Fixes in code for v0.2 * Add new requirements, modify conf etc.
1 parent 2b15cbb commit adcb718

File tree

7 files changed

+13
-6
lines changed

7 files changed

+13
-6
lines changed

.gitignore

+3
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,12 @@ advanced
55

66
#data things
77
beginner_source/hymenoptera_data
8+
beginner_source/blitz/data
9+
beginner_source/faces
810
intermediate_source/data/
911
advanced_source/images/
1012
*data.zip
13+
faces.zip
1114

1215
#builds
1316
_build/

beginner_source/blitz/cifar10_tutorial.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def forward(self, x):
216216
# So, let's get the index of the highest energy:
217217
_, predicted = torch.max(outputs.data, 1)
218218

219-
print('Predicted: ', ' '.join('%5s' % classes[predicted[j][0]]
219+
print('Predicted: ', ' '.join('%5s' % classes[predicted[j]]
220220
for j in range(4)))
221221

222222
########################################################################

beginner_source/transfer_learning_tutorial.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def visualize_model(model, num_images=6):
253253
images_so_far += 1
254254
ax = plt.subplot(num_images//2, 2, images_so_far)
255255
ax.axis('off')
256-
ax.set_title('predicted: {}'.format(dset_classes[preds[j][0]]))
256+
ax.set_title('predicted: {}'.format(dset_classes[preds[j]]))
257257
imshow(inputs.cpu().data[j])
258258

259259
if images_so_far == num_images:

conf.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,8 @@
5454
'examples_dirs': ['beginner_source', 'intermediate_source',
5555
'advanced_source'],
5656
'gallery_dirs': ['beginner', 'intermediate', 'advanced'],
57-
'filename_pattern': 'tutorial.py'
57+
'filename_pattern': 'tutorial.py',
58+
'backreferences_dir': False
5859
}
5960

6061
for i in range(len(sphinx_gallery_conf['examples_dirs'])):

index.rst

-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ many other applications at https://github.com/pytorch/examples/
1616

1717
You can find reference documentation for PyTorch's API and layers at
1818
http://docs.pytorch.org or via inline help.
19-
2019
If you would like the tutorials section improved, please open a github issue
2120
here with your feedback: https://github.com/pytorch/tutorials
2221

intermediate_source/seq2seq_translation_tutorial.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -592,7 +592,7 @@ def train(input_variable, target_variable, encoder, decoder, encoder_optimizer,
592592
for di in range(target_length):
593593
decoder_output, decoder_hidden, decoder_attention = decoder(
594594
decoder_input, decoder_hidden, encoder_output, encoder_outputs)
595-
loss += criterion(decoder_output[0], target_variable[di])
595+
loss += criterion(decoder_output, target_variable[di])
596596
decoder_input = target_variable[di] # Teacher forcing
597597

598598
else:
@@ -606,7 +606,7 @@ def train(input_variable, target_variable, encoder, decoder, encoder_optimizer,
606606
decoder_input = Variable(torch.LongTensor([[ni]]))
607607
decoder_input = decoder_input.cuda() if use_cuda else decoder_input
608608

609-
loss += criterion(decoder_output[0], target_variable[di])
609+
loss += criterion(decoder_output, target_variable[di])
610610
if ni == EOS_token:
611611
break
612612

requirements.txt

+4
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,7 @@ numpy
55
matplotlib
66
torchvision
77
torch
8+
9+
# to run examples
10+
pandas
11+
scikit-image

0 commit comments

Comments
 (0)