topaz.protocols.protocol_training module