Browse Source

Update tutorial_5_datasetiter.rst

tags/v0.4.10
zide05 GitHub 5 years ago
parent
commit
cd395a7cdf
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      docs/source/tutorials/tutorial_5_datasetiter.rst

+ 1
- 1
docs/source/tutorials/tutorial_5_datasetiter.rst View File

@@ -192,7 +192,7 @@ sampler
import time

embed_dim = 100
model = CNNText((len(vocab),embed_dim), num_classes=3, padding=2, dropout=0.1)
model = CNNText((len(vocab),embed_dim), num_classes=3, dropout=0.1)

def train(epoch, data, devdata):
optimizer = torch.optim.Adam(model.parameters(), lr=0.001)


Loading…
Cancel
Save