From 1511778267a964d261c7bbf9f15233cb92adf3ab Mon Sep 17 00:00:00 2001 From: yuebanyishenqiu Date: Sun, 22 Mar 2020 09:41:29 +0800 Subject: [PATCH] fix typos --- src/data/dataloader.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/data/dataloader.jl b/src/data/dataloader.jl index 9da14650..07c8f1fd 100644 --- a/src/data/dataloader.jl +++ b/src/data/dataloader.jl @@ -30,7 +30,7 @@ Example usage: Xtrain = rand(10, 100) train_loader = DataLoader(Xtrain, batchsize=2) # iterate over 50 mini-batches of size 2 - for x in train_loader: + for x in train_loader @assert size(x) == (10, 2) ... end @@ -41,7 +41,7 @@ Example usage: Ytrain = rand(100) train_loader = DataLoader(Xtrain, Ytrain, batchsize=2, shuffle=true) for epoch in 1:100 - for (x, y) in train_loader: + for (x, y) in train_loader @assert size(x) == (10, 2) @assert size(y) == (2,) ... @@ -89,4 +89,4 @@ end function Base.length(d::DataLoader) n = d.nobs / d.batchsize d.partial ? ceil(Int,n) : floor(Int,n) -end \ No newline at end of file +end