Merge pull request #10 from mashu/dg/data

Relax type of returned values by DataLoader i…
DhairyaLGandhi · Dec 10, 2021 · 14645b4 · 14645b4
2 parents cff2999 + 3166552
commit 14645b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/data/dataloader.jl b/src/data/dataloader.jl
@@ -128,7 +128,7 @@ function DataLoader(f,
   fs = map(feat -> getindex(feat,
                  ntuple(i -> i == batchdim(feat) ? ix : Colon(), length(size(feat)))...), args)
   iterator = Iterators.partition(ix, batchsize)
-  ch = Channel{typeof(args)}(buffersize)
+  ch = Channel(buffersize)
   t = Task(() -> begin
     for i in iterator
       fullbatch = length(i) == batchsize