minor fixes to multi-gpu training

[krisrak]

Existing Sample Changes

Description

minor fixes to multi-gpu training, updated to support 8 GPUs in new IDC

Type of change

• [X] Bug fix (non-breaking change which fixes an issue)

How Has This Been Tested?

• [X] Command Line