pytorch下用UCF101数据集仿真C3D模型
上次想用老旧thinkpad装linux系统跑神经网络的想法实在是太愚蠢了,因为在从github上下了代码一番折腾(解决飘红报错)后发现这个19年买的灵越7000根本也跑不起神经网络的代码。最后运行报错:RuntimeError: CUDA out of memory. Tried to allocate 148.00 MiB (GPU 0; 2.00 GiB total capacity;979
上次想用老旧thinkpad装linux系统跑神经网络的想法实在是太愚蠢了,因为在从github上下了代码一番折腾(解决飘红报错)后发现这个19年买的灵越7000根本也跑不起神经网络的代码。
最后运行报错:
RuntimeError: CUDA out of memory. Tried to allocate 148.00 MiB (GPU 0; 2.00 GiB total capacity;
979.11 MiB already allocated;38.88 MiB free;
1.04 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
大意了。。。。找了实验室学长,用实验室的服务器,pycharm可以上传代码运行,远程浏览文件夹目录等,但我还发现了一个好东西!MobaXterm,用linux指令,挺方便,还有Portable免安装版。
代码是GitHub上搜C3D的代码,对照唐宇迪老师的视频学习
https://www.bilibili.com/video/BV1Mu411d7Jt?spm_id_from=333.999.0.0
原代码eporch=51左右,loss和acc就趋于平稳了,感觉识别效果一般,很多动作识别都是错误的,据b站小伙伴说是模型较老,唐老师说他参数没有细调,因为我的动作识别必须要自己的数据集自己的动作,这个只有UCF101,代码我新手改起来也好吃力,所以这个暂时搁置,去学习别的模型了。

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)