-
pytorch每个批次保存一次网络的写法:
torch.save(net.state_dict(), ‘weights/ yolo_%d.pt’ % epoch) -
判断文件是否存在,若不存在,则创建
os.path.exists()与 os.path.isdir()两个都可以
区别:
os.path.exists():只判断文件是否存在
os.path.isdir():isdir可以理解为is directory,它不仅关心文件是否存在,也关注文件的类型
通俗来讲,若是一个txt文件,则exists可行,但是isdir不可,因为txt是一个文本
完整代码如:
``
这里:mkdir与makedirs的区别:
mkdir:创建一级目录
makedirs:可以创建二级目录