YOLO 学習の再開方法

Question

YOLOを用いて学習をしているのですが、Google Colaboratoryを使用しているため一定時間が経過すると，PUが使用できなくなり、学習が停止してしまいます．
例えば20epochまで学習が終了している場合、次に学習を再開する際に21epochから学習を始めるにはどのようにすれば良いでしょうか．

Answer 1

--resumeというオプションと、途中までの学習で生成されたlast.ptを用いることで、学習を再開することができました．

Answer 2

[Google Colab] 学習を中断した後、途中から再開する方法

↑の記事が参考になると思います。

以下、蛇足です

多分Colabって環境自体が整ってますけど利用制限がある都合上、全くのド素人が触り始めるには
向いてるけど、本格的に勉強したいってなると制限の都合上、学習がストップしてしまうので
それだったら自前で環境構築した方が勉強にもなるし良いんじゃないかなと思いました。

Comments

[yamiyami]

コメントいただきありがとうございます．
Kerasではmodel.fitを用いて学習しているようですが、当方の使用しているYOLOv5ではどこにcallbacks=[checkpoint]などを記載すれば良いのでしょうか．
コードの読み書きについてはまだ素人なので、教えていただけたら幸いです．

また、YOLOv5で学習するにあたり、--nosave (only save final checkpoint)というオプションを見つけましたが、このオプションを外すことで、常にチェックポイントを生成できるのでしょうか．
この場合、生成されたチェックポイントはどのように使用すれば良いのでしょうか．
複数伺ってしまい、申し訳ありません。

K_3578様の仰る通り、colabでは少し限界を感じてきています．
AWSも少し気にはなっているのですが、追々、自前の環境を構築したいと考えています．

[K_3578]

あー、すいません気付いてなかったんですが解決されたようで何よりです。
一応teratailはスレッド一つにつき質問は一つとして頂きたいので追加で質問などある場合は新規スレッドを建てられてください。

そうですね、その辺の環境構築なんかも機械学習なんかは人気のあるコンテンツなので
Qiita等記事が沢山有ると思うので参考にしてください。

勿論環境構築についてteratailで質問するのは構いませんが、色々詳細が書かれていると
回答されやすくなると思います。

お役に立てなかったようで申し訳ない。