PyTorchのargparse直設定方法

Question

PyTorchの実行ファイルで、引数受け取りにargparseを使いますが、これをargparseクラスを使いつつ、プログラム内で直に設定する方法を教えてください。
普通は

python train.py --epoch 1000

としますが、これをtrain.pyを呼ばず直接

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--epoch', type=int, default=0) #ここで直に設定したい
opt = parser.parse_args()

としたいです。

なぜかというと、JupyterNotebookのセル上で、デバッグさせたいからです。

上記コードをセルで実行すると、

usage: ipykernel_launcher.py [-h] [--epoch EPOCH]
ipykernel_launcher.py: error: unrecognized arguments: -f /root/.local/share/jupyter/runtime/kernel-7f25afe7-fcc9-406e-874c-7c60d459d8d6.json
An exception has occurred, use %tb to see the full traceback.

SystemExit: 2
/usr/local/lib/python3.6/dist-packages/IPython/core/interactiveshell.py:2890: UserWarning: To exit: use 'exit', 'quit', or Ctrl-D.
  warn("To exit: use 'exit', 'quit', or Ctrl-D.", stacklevel=1)

というエラーになります。
opt = parser.parse_args()
の部分でインスタンスのメソッドを代入できないのです。

Answer 1

https://qiita.com/uenonuenon/items/09fa620426b4c5d4acf9
はどうでしょう？

Comments

[teefpc]

#opt = parser.parse_args()
opt = parser.parse_args(args=[])
で動きました。ありがとうございました。

Answer 2

defaultに「直に設定したい値」を入れるのではダメなのでしょうか？

Comments

[teefpc]

はい。一応、インスタンスに格納されているようですが、取り出そうとすると上記追加文のようにエラーとなります。