dict etc to help with restart of the training                  
                  <b>torch.save(model_param_dict,output_model_file_name.format(step))</b>
                  
               <b>if step % conf.display_step == 0:</b>
                  # every display_step save training information details. This information is displayed by APT front end in training monitor GUI during training.
                  train_loss = loss.cpu().numpy() 
                  # to ensure that the GPU memory gets collected by th