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 An object that iterates over mini-batches of `data`, each mini-batch containing `batchsize` observations
(except possibly the last one). (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. 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. If `shuffle=true`, shuffles the observations each time iterations are re-started.