--- datasets: - edinburghcstr/ami --- Deit model supcon + GE2E learning Ami dataset