Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -309,7 +309,7 @@ Few pointers to get you started:

- [Quick Start Guide: Essentials of getting a project up and running](https://pytorch.org/ignite/quickstart.html)
- [Concepts of the library: Engine, Events & Handlers, State, Metrics](https://pytorch.org/ignite/concepts.html)
- Full-featured template examples for common tasks (coming soon ...)
- Full-featured template examples (coming soon ...)


<!-- ############################################################################################################### -->
Expand Down
6 changes: 1 addition & 5 deletions examples/contrib/cifar10/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,11 +196,7 @@ def get_dataflow(config):
)

test_loader = idist.auto_dataloader(
test_dataset,
batch_size=2 * config["batch_size"],
num_workers=config["num_workers"],
shuffle=False,
pin_memory="cuda" in idist.device().type,
test_dataset, batch_size=2 * config["batch_size"], num_workers=config["num_workers"], shuffle=False,
)
return train_loader, test_loader

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@ def train_step(engine, batch):
y_pred = model(x)
loss = criterion(y_pred, y)

# Runs the forward pass with autocasting.
with amp.scale_loss(loss, optimizer) as scaled_loss:
scaled_loss.backward()

Expand Down