TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Apply suggestions from code review 

accept suggested changes

Co-authored-by: Carlo Lucibello <carlo.lucibello@gmail.com>
This commit is contained in:
cossio 2020-06-12 11:45:42 +02:00 committed by cossio
parent 909a55ac10
commit 75692161a7
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/data/dataloader.jl
View File

@ -16,7 +16,7 @@ end
An object that iterates over mini-batches of `data`, each mini-batch containing `batchsize` observations
(except possibly the last one).
Takes as input a data tensors or a tuple (or `NamedTuple`) of one or more such tensors.
Takes as input a single data tensor, or a tuple (or a named tuple) of tensors.
The last dimension in each tensor is considered to be the observation dimension.
If `shuffle=true`, shuffles the observations each time iterations are re-started.