search

ecm200 / caltech_birds

A set of notebooks as a guide to the process of fine-grained image classification of birds species, using PyTorch based deep neural networks.
41 stars 7 forks source link

Make command line argument configuration overrides work by using ARGS module in CUB_TOOLS package #33

Open ecm200 opened 3 years ago

ecm200 commented 3 years ago

At the moment both the base and ignite training scripts are not using the args module of the cub_tools package.

This has certain advantages, like passing through command line arguments key value pairs to override the YACS YAML configuration file.

Use the following code to in the scripts to create the argument parser with some default arguments:

from cub_tools.args import get_parser

parser = get_parser()
parser.add_argument # add additional local arguments to this specific training script if needed.

args = parser.parse_args()