摘要:
torchvision中提供了很多训练好的模型,这些模型是在1000类,224*224的imagenet中训练得到的,很多时候不适合我们自己的数据,可以根据需要进行修改。 1、类别不同 2、添加层后,加载部分参数 参考:https://blog.csdn.net/u012494820/article 阅读全文
摘要:
yield可将一个函数变成生成器,每次调用时,返回yield的结果,下次迭代时,从yield 下条语句开始执行。 一个典型的例子,斐波拉切数列: 输出 参考:https://www.liaoxuefeng.com/article/001373892916170b88313a39f294309970a 阅读全文