Help me identify a problem in Machine Learning Code

I need an expert in computer vision, who can help me in identifying a very basic problem. I am trying to run a Semantic Segmentation code from github on a docker. The code uses pytorch. However, the results after training are not accurate. Need someone with good experience to find the cause of problem. The code is 100% correct. The issue lies in the environment setup.