Function save_checkpoint

code/DeepDG/utils/util.py:21–26

(filename, alg, args)

def save_checkpoint(filename, alg, args):
    save_dict = {
        "args": vars(args),
        "model_dict": alg.cpu().state_dict()
    }
    torch.save(save_dict, os.path.join(args.output, filename))


def train_valid_target_eval_names(args):