Training fixes & early stop

• 074ea02 bugfix with logger and tqdm bar state, could result in encoding errors
• 18f5fdf training now handles early stop, to specify in iterator_kwargs
• 2d114a5 device_ arg renamed to device in train method