Framework - Adding pytorch profiler

[punit121]

When I try to run the profiler on train_net.py within my new conda environment, it fails because it can't load the COCO dataset. It seems like there's a problem with how my code is trying to find and use the dataset.