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.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,代码我新手改起来也好吃力,所以这个暂时搁置,去学习别的模型了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值