Improve grammar (#1619)

This commit is contained in:
Gadersd 2024-04-16 09:32:42 -04:00 committed by GitHub
parent adb31dd943
commit 1235b06e25
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions

View File

@ -125,7 +125,7 @@ pub fn train<B: AutodiffBackend>(artifact_dir: &str, config: TrainingConfig, dev
B::seed(config.seed);
let batcher_train = MnistBatcher::<B>::new(device.clone());
let batcher_valid = MnistBatcher::<B::InnerBackend>::new(device.clone());
let batcher_valid = MnistBatcher::<B>::InnerBackend>::new(device.clone());
let dataloader_train = DataLoaderBuilder::new(batcher_train)
.batch_size(config.batch_size)

View File

@ -1,6 +1,6 @@
# Dataset
Most deep learning training being done on datasets with perhaps the exception of reinforcement learning, it is
In most deep learning training performed on datasets (with perhaps the exception of reinforcement learning), it is
essential to provide a convenient and performant API.
The dataset trait is quite similar to the dataset abstract class in PyTorch:
@ -69,7 +69,7 @@ let dataset = ShuffledDataset<DbPedia, DbPediaItem>::with_seed(dataset, 42);
```rust, ignore
// define chained dataset type here for brevity
type PartialData = PartialDataset<ShuffledDataset<DbPedia, DbPediaItem>>;
let dataset_len = dataset.len();
let len = dataset.len();
let split == "train"; // or "val"/"test"
let data_split = match split {