pre_trained_checkpoint_4ch