You should never train with a collar on a yorkie. Even a body harness that goes across their chest is harmful. Use a "step in" harness for walks. If you are training you can use one of the muzzle trainers. We use it and it works. Your leash is attached under the chin, so it never chokes them.
Good luck.