vak icon indicating copy to clipboard operation
vak copied to clipboard

make it possible for predict to load very large spectrograms

Open NickleDave opened this issue 4 years ago • 2 comments

will need to turn them into multiple batches, probably within predict function?

or maybe the data loader can take care of this using a clever collate_fn

NickleDave avatar Oct 01 '20 15:10 NickleDave

since we're already using ViewAsWindowBatch, quick-and-dirty solution might be to set a max_batch_size attribute and then split above that. Then let the collate_fn deal with those splits (?)

NickleDave avatar Oct 01 '20 15:10 NickleDave

You should be able to reuse the data generator you use for training, no? (Without shuffling ofc).

atomfried avatar Oct 02 '20 13:10 atomfried