我花了5天时间配置亚马逊网络服务SageMaker。(联系增加GPU笔记本限制需要2天,下载解压所有数据需要2天,上载数据到S3等绕道需要1天)。
当我终于可以在SageMaker GPU笔记本实例上运行代码时,我发现速度并不比我本地的CPU快。
我使用下面的代码来检查GPU。不知道哪里出了问题。这会发生在你们这些朋友身上吗?而我的学分只剩下200美元(原来是750美元)。
发布于 2020-02-18 18:20:12
问题应该出在视频加载上。当我在图像数据集上训练一个Resnet模型时,速度相当快!!感谢AWS !!
关于视频加载策略,我制作了这个内核来比较不同的方法。
https://www.kaggle.com/muerbingsha/deepfake-face-detector-evaluate
事实证明imageio_ffmpeg
是最快的方法。
https://stackoverflow.com/questions/60243228
复制相似问题