fix bug where train size and monitor size had to be the same