最近在准备机器人竞赛,憨憨机器人需要完成目标检测。大家知道,训练目标检测的模型还是非常吃资源的。有可能模型训练结束后,电脑也被榨干了。于是,在学长的建议下,我决定使用google colab进行模型的训练。美其名曰尝试新事物,实则白嫖GPU,怎一个“香”字了得~
但是,在使用中也会遇到一些小问题,比如,colab自动断连~~~
这其实比较好解决,因为只要动动手指就行了。但是,如果发现得不及时,比如训练1W次左右往往需要很久很久,这时候你打算先前往峡谷征战一番,待尔大杀四方凯旋而归时,打开心爱的colab,发现训练次数停留在了1256次。。。。。。(不要问我为什么知道这个准确数字(꒦_꒦) )
但是客观情况下确实不可能守在电脑前7、8个小时,只为了防止某一瞬间的掉线。所以,本篇博客应运而生。
参考这个链接可以得到一些答案。
具体来说,colab理论上最多可以连续工作12小时,这之后就会再重新分配资源,但这之中也会存在因为网络原因或者过久没有交互而导致的disconnect的情况,因此上述内容主要是为了解决这种问题的。但是如果训练的时间本身就是超过12小时所以断开了,那就不适用了。大家也别急着喷,毕竟白嫖的嘛。
以下是我最近使用的方法:
首先调出网页控制台(Console) Ctrl+Shift+i
或按 F12
均可找到:
然后在控制台内输入以下代码,敲回车键:
function ConnectButton(){
console.log("Connect pushed");
document.querySelector("#top-toolbar > colab-connect-button").shadowRoot.querySelector("#connect").click()
}
setInterval(ConnectButton,60000);
以上代码会每隔60面自动“push”一次与colab交互
这样,就可以放心地在峡谷征战啦!✿✿ヽ(°▽°)ノ✿
然后得到训练结果- - - -
参考
[1].https://stackoverflow.com/questions/57113226/how-to-prevent-google-colab-from-disconnecting
[侵删]
预告
下一篇《Q-learning、Expected Sarsa、Double Learning》,对强化学习感兴趣的小伙伴可以一起交流噢!