Updated code (contrastive learning) (#130)

* updated prediction code to test specific cells

* updated kernel and stride

* updated kernel and stride

* new training script and predict takes in output path as parameter

* passing ci requirements

* fixed error for output path

* fixed error for output path

* fixed formatting

* format resolved

* docstrings for methods that parse tracks

---------

Co-authored-by: Shalin Mehta <[email protected]>

Author: 2 people