-
Geoffrey Yu authored
fixed the input data generation so that true_msa and bert_mask shapes are correct otherwise crashes in loss calculations
1b4fc115
fixed the input data generation so that true_msa and bert_mask shapes are correct otherwise crashes in loss calculations