「ヒドラで実験を追跡し続けましょう」

Let's continue tracking experiments in Hydra.

(Image by Author)

YAMLファイルを使用してハイパーパラメーターを設定し、研究をスピードアップしましょう!

はじめに

コードにバグがないものを最初の試行で書くことは不可能であるように、最初の試行で正しいモデルをトレーニングすることも不可能です。

機械学習やディープラーニングの経験がある方は、モデルの適切なハイパーパラメーターを選ぶのに多くの時間を費やす必要があることを知っています。これらのハイパーパラメーターには、学習率、バッチサイズ、出力のクラス数などがありますが、これらは一部の一般的なものであり、プロジェクトには数百ものパラメーターが存在する場合もあります。

ハイパーパラメーターを変更することで、異なる結果(良いものや悪いもの)が得られる可能性があり、ある時点で行ったすべてのテストを追跡するのは非常に困難です。

私が長い間行っていた方法は次のとおりです:これらのハイパーパラメーターを手作業でエクセルシートに書き留め、各実験の結果、例えば損失値を隣に書き留めるというものでした。後に「進化」し、ハイパーパラメーターのための設定ファイルを作成し、テストしたいさまざまな値を記入しました。後で、それらの値を読み取り、トレーニング関数に挿入するためのカスタムPython関数を作成していました。YAMLファイルは基本的には階層的に構築されたファイルであり、次のようなキーと値を挿入することができます:

data:  path: "data/ESC-50"  sample_rate: 8000  train_folds: [1, 2, 3]  val_folds: [4]  test_folds: [5]  batch_size: 8model:  base_filters: 32  num_classes: 50  optim:    lr: 3e-4seed: 0trainer:  max_epochs: 10

後に、この全プロセスを簡単化し、さらに速くするオープンソースのフレームワークであるHydraを見つけました。

さあ、始めましょう!

PyTorchを使用してシンプルな機械学習プロジェクトを開発しているとします。通常通り、データセットのためのクラスを作成し、データローダーをインスタンス化し、モデルを作成してトレーニングします。この例では、コードをより整理するためにPyTorch Lightningを使用し、Kerasで行うようなTrainerオブジェクトを持つことにします。PyTorchに慣れている場合は、Lightningもすぐに理解できるでしょう。

We will continue to update VoAGI; if you have any questions or suggestions, please contact us!

Share:

Was this article helpful?

93 out of 132 found this helpful

Discover more

人工知能

「マーク・A・レムリー教授による生成AIと法律について」

データサイエンス内で新しい分野が現れ、研究内容が理解しにくい場合は、専門家やパイオニアと話すことが最善です最近、私た...

人工知能

『DeepHowのCEO兼共同創業者、サム・ジェン氏によるインタビューシリーズ』

ディープハウのCEO兼共同創設者であるサム・ジェンは、著名な投資家から支持される急速に進化するスタートアップを率いていま...

人工知能

「スノーケルAIのCEO兼共同創設者、アレックス・ラットナー - インタビューシリーズ」

アレックス・ラトナーは、スタンフォードAIラボを母体とする会社、Snorkel AIのCEO兼共同創設者ですSnorkel AIは、手作業のAI...

人工知能

「トリントの創設者兼CEO、ジェフ・コフマンへのインタビューシリーズ」

ジェフ・コーフマンは、ABC、CBS、CBCニュースで30年のキャリアを持った後、Trintの創設者兼CEOとなりましたジェフは手作業の...

人工知能

Aaron Lee、Smith.aiの共同設立者兼CEO - インタビューシリーズ

アーロン・リーさんは、Smith.aiの共同創業者兼CEOであり、AIと人間の知性を組み合わせて、24時間365日の顧客エンゲージメン...

機械学習

「機械学習 vs AI vs ディープラーニング vs ニューラルネットワーク:違いは何ですか?」

テクノロジーの急速な進化は、ビジネスが効率化のために洗練されたアルゴリズムにますます頼ることで、私たちの日常生活を形...