|
|
@@ -155,7 +155,7 @@ class DatasetHelper: |
|
|
>>> network = Net() |
|
|
>>> network = Net() |
|
|
>>> net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean") |
|
|
>>> net_loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean") |
|
|
>>> network = nn.WithLossCell(network, net_loss) |
|
|
>>> network = nn.WithLossCell(network, net_loss) |
|
|
>>> train_dataset = create_custom_dataset() |
|
|
|
|
|
|
|
|
>>> train_dataset = create_custom_dataset(sparse=True) |
|
|
>>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False) |
|
|
>>> dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False) |
|
|
>>> for next_element in dataset_helper: |
|
|
>>> for next_element in dataset_helper: |
|
|
... outputs = network(*next_element) |
|
|
... outputs = network(*next_element) |
|
|
|