Fix train_requirements.txt to run as currently pytorch gives an error and it stops all subsequent packages from being downloaded
Fix train_requirements.txt to run as currently pytorch gives an error and it stops all subsequent packages from being downloaded