andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Morvan Zhou
2017-05-23 10:37:51 +10:00
committed by Morvan Zhou
parent 30c0070497
commit d87529ed89
10 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorial-contents/504_batch_normalization.py
View File

@ -48,7 +48,6 @@ train_loader = Data.DataLoader(dataset=train_dataset, batch_size=BATCH_SIZE, shu
# show data
plt.scatter(train_x.numpy(), train_y.numpy(), c='#FF9359', s=50, alpha=0.2, label='train')
plt.legend(loc='upper left')
plt.show()
class Net(nn.Module):
def __init__(self, batch_normalization=False):