Remove unecessary stuff in train_network script

It looks like this line is not needed: https://gitlab.idiap.ch/bob/bob.learn.pytorch/blob/master/bob/learn/pytorch/scripts/train_network.py#L322