Patch 1

hq615requested to merge
(removed):patch-1 into master

4 commits bring the following changes:

  1. Fixed the dependency issue with torchsample, see changes in requirements.txt. This package is not included in PyPI library for pip3, hence pip install torchsample did not work. pandas, nibabel and tqdm are also added since they are required by torchsample.

  2. Added GPU selection support in predict.py --gpu using shell environment masking method. Not the neatest way but works.

  3. Fixed the typo in the instruction for running prediction in ReadME.md.

  4. Applied simple automated PEP-8 style code reformatting for better readability.

Merge request reports