通用视觉框架OpenMMLab 第三课速通版
评价
这节课的内容主要在讲代码和框架的使用。 本节课的内容很少,除去北京超算平台的使用讲解(我使用学院服务器,这部分就直接没听),课程分为两阶段,第一阶段是对mmcls框架使用的简单介绍,覆盖性的介绍了一遍。然后就是同济子豪进行答疑。
值得记录的点
-
图像输入的大小是否固定? 答:这个问题的根源在于网络是否使用全连接层。全连接层的使用,倒逼输入尺寸的大小。比如MobileNetV2使用了全局平均池化,而不是全连接层,就可以适配任何尺寸的图片输入。
-
一个查看模型参数量和运算量的脚本。 在tools/analysis_tools下面有一个get_flops.py的脚本,用python执行,配置文件作为args参数。即可打印对模型的粗略的计算。
-
lastest 是一个软连接,你可以在配置文件直接引用lastest的路径,但是不要擅自删除真实的pth。不想要旧版本的pth,看清楚再删。这个问题我很清楚但还是会犯。
疑惑
让我疑惑的是,我在训练中指定了两卡训练,但是一看GPU使用情况,却是单卡卡0在跑。 我的指定方法如下。如果哪里有什么问题欢迎帮忙指出,谢谢。
python ./tools/train.py --gpus 2 ${config}