primeqa.ir.dense.colbert_top.colbert.training.utils.manage_checkpoints_with_path_save

primeqa.ir.dense.colbert_top.colbert.training.utils.manage_checkpoints_with_path_save(args, colbert, optimizer, amp, batch_idx, num_per_epoch, epoch_idx=0, train_loss=0)