Run this:

 python tac_ssl_tag.py --dataset pretrain