import torch
import shutil

def save_checkpoint(state, is_best: bool, task_id, fname="checkpoint.pth.tar"):
    fdir = "./"+str(task_id)+"/"
    torch.save(state, fdir + fname)
    if is_best:
        shutil.copyfile(fdir + fname, fdir + "best.pth.tar")