I am Charmie

メモとログ

2023-10-01から1ヶ月間の記事一覧

PyTorch Lightning + Hydra

PyTorch LightningとHydraを組み合わせるという流れに乗ろうと思った. このレポジトリで公開しているテンプレートが便利 WanDBを使ったダッシュボードは面白かった 以下がパッと分からなかったので導入は新年度からにする WanDB 各グラフを画像として取得す…

PyTorch Lightning

PyTorch Lightningを使ってみた. この解説がわかりやすかった PyTorchにおけるBoilerplate codeを減らすための工夫が施されている こんな感じで対応してる. 学習の処理(一バッチ分)をモデルクラスの関数に書くことで,学習の二重ループを書かなくて済む …

Ubuntu 22.04: GPU on docker

ここが参考になる Docker上でGPUを使う場合はCUDAのインストールは不要 nvidia-docker)の立ち位置がわからない... NVIDIA Container Toolkitのこと? NVIDIA Container Toolkitのインストール後に自分のdockerfileはどんな風に書いたらいいのか調べる Docker…