sainivedh19pt

Results 3 issues of sainivedh19pt

- Added Dockerfile and minor code changes to support latest pytorch - Added multi batch inference support in active_learning_cycle

For training on custom dataset, suppose stl-10..should the input be flattened(3x96x96) as here mnist we're passing(784,) vector

Thanks for your amazing work. **Error** ```tensorflow.python.framework.errors_impl.InvalidArgumentError: Name: , Key: image/encoded, Index: 0. Number of values != expected. values size: 100 but output shape: [] [[{{node ParseSingleSequenceExample/ParseSequenceExample/ParseSequenceExampleV2}}]] [Op:IteratorGetNext] 2022-04-15 13:19:55.258721:...