torch.utils.checkpoint.checkpoint (function, *args, **kwargs)[source]. 检查模型或者模型的一部分。通过将计算变为内存来进行检查点工作。而不是存储 ...
確定! 回上一頁