こすたろーんエンジニアの試行錯誤部屋

作成物の備忘録を書いていきますー

abseilで複数の引数をflagfileで一括設定する

目次

スポンサーリンク

この記事でわかること

abseilとテキストファイルを使った引数処理方法

1.実行環境

Jetson Xavier NX
ubuntu18.04
docker
python3.x
pytorch
->Jetson Xavier NX上におけるpytrorch環境構築は以下でやってますので、ご参考までに~
technoxs-stacker.hatenablog.com

2.基本的な使い方

2.1テキストファイルに引数設定を記述
2.2python実行時にテキストファイルを指定する

2.1テキストファイルに引数設定を記述

--arch=resnet50
--gpu=0
--batch_size=8
--print_freq=10
--pred_dim=256

2.2python実行時にテキストファイルを指定する

CUDA_VISIBLE_DEVICES=0 python3 main_simsiam_single.py --flagfile ./config/simsiam.txt

関連記事

technoxs-stacker.hatenablog.com

感想

複数の引数を実行時に毎回記述するのはめんどくさいので、flagfileで指定できるのは
非常に助かります

スポンサーリンク

参考

abseil.io xvideos.hatenablog.com github.com