removing logging from trainer