`batch_size` and `n_updates` are hardcoded in `training_script.py`

[jbirkesteen]

These variables are provided as arguments but their values are overwritten in the implemented function.

[jbirkesteen]

Also, in the same script: the run.stop() call on line 78 needs to be guarded by an if-clause, since run is created only if logging is activated.