hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

update

tags/v0.1.0
HENRY L 6 years ago
parent
561305e03d
commit
f585a9aa7d
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      fastNLP/modules/prototype/example.py

+ 1
- 1
fastNLP/modules/prototype/example.py View File

@@ -60,7 +60,7 @@ def train(model_dict=None, using_cuda=True, learning_rate=0.06,\


optimizer = optim.SGD(net.parameters(), lr=learning_rate, momentum=momentum) optimizer = optim.SGD(net.parameters(), lr=learning_rate, momentum=momentum)
criterion = nn.CrossEntropyLoss() criterion = nn.CrossEntropyLoss()
dataset = dataloader.DataLoader("test_set.pkl", batch_size, using_cuda=using_cuda)
dataset = dataloader.DataLoader("train_set.pkl", batch_size, using_cuda=using_cuda)


#statistics #statistics
loss_count = 0 loss_count = 0


Write Preview
Loading…
Cancel
Save