segmentation.py 文件源码

def save_param(out_dir, epoch, storage):
    serializers.save_npz(
        str(out_dir/model_name(epoch)),
        storage.model
    )
    serializers.save_npz(
        str(out_dir/optimizer_name(epoch)),
        storage.optimizer
    )