primeqa.ir.dense.colbert_top.colbert.utils.utils.save_checkpoint

primeqa.ir.dense.colbert_top.colbert.utils.utils.save_checkpoint(path, epoch_idx, mb_idx, model, optimizer, amp, train_loss)