modifying inner training loop of YNet

3 jobs for master in 2 minutes and 24 seconds (queued for 1 second)
Status Job ID Name Coverage
  Test
passed #291171
build

00:00:52

passed #291172
pages

00:00:32

failed #291170
test

00:01:00

 
Name Stage Failure
failed
test Test

WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available.
Use 'nvidia-docker run' to start this container; see
https://github.com/NVIDIA/nvidia-docker/wiki/nvidia-docker .

NOTE: MOFED driver for multi-node communication was not detected.
Multi-node communication performance may be reduced.

ERROR: Job failed: exit code 1