search

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
7.68k stars 2.53k forks source link

我想保存模型结构和参数 #3707

Open LiangYong1216 opened 2 weeks ago

LiangYong1216 commented 2 weeks ago

我想保存整个模型结构和参数，而不是仅仅保存state_dict,我需要如何配置呢？需要修改哪些文件呢？