Worse performance when executing model.eval() than model.train()