search

pesser / edflow

Reduce boilerplate code for your ML projects. TensorFlow and PyTorch.
https://edflow.readthedocs.io/
MIT License
24 stars 13 forks source link

adds get_fixed_examples to TemplateIterator #268

Closed theRealSuperMario closed 4 years ago

theRealSuperMario commented 4 years ago
  • I think the interface would be more consistent if the function just returns a batch exactly the way the step_op receives the batch instead of having to specify the keys. In addition it would be great if we can make sure that the behavior is exactly that of the dataloader - I think this is readily achieved by using iterators.batches.deep_lod2dol

Done

theRealSuperMario commented 4 years ago

it would be great if one can specify which dataset to take the examples from, eg train or validation.

Done

pesser commented 4 years ago

Great! Now I can finally get rid of the copy & pasted code in my iterators :) :+1: