Update requirements.txt

[UncleBob2]

getting all the detailed requirements to get the environment setup for training.

[v-nhandt21]

@UncleBob2 Should we install torch==1.6.0 and torchvision==0.7.0 from Pytorch Main page instead of pypi :)) I think for others GPU architecture may require different torch version :))