Bunch-Training-retrieval

[daveabiy]

1. Next task:- Training a bunch of images at once
   It doesn't necessarily need to save and init the model, but parallel training DataLoader class in pytorch #4

2. Adding separately trained models and save them (as an average prediction).

3. See if the artifacts created by predictions can be removed