From 08762b32cf73a985abcc58291a27b290f942e1cd Mon Sep 17 00:00:00 2001 From: jsadler2 Date: Tue, 15 Jun 2021 14:01:37 -0500 Subject: [PATCH] [#111] train_model.py -> train_model_cli.py --- Snakefile | 2 +- river_dl/{train_model.py => train_model_cli.py} | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) rename river_dl/{train_model.py => train_model_cli.py} (90%) diff --git a/Snakefile b/Snakefile index d2a9c1b..e8a0256 100644 --- a/Snakefile +++ b/Snakefile @@ -59,7 +59,7 @@ rule prep_io_data: # shell: # """ # module load analytics cuda10.1/toolkit/10.1.105 -# run_training -e /home/jsadler/.conda/envs/rgcn --no-node-list "python {code_dir}/train_model.py -o {params.run_dir} -i {input[0]} -p {params.pt_epochs} -f {params.ft_epochs} --lambdas {params.lamb} --loss_func multitask_rmse --model rgcn -s 135" +# run_training -e /home/jsadler/.conda/envs/rgcn --no-node-list "python {code_dir}/train_model_cli.py -o {params.run_dir} -i {input[0]} -p {params.pt_epochs} -f {params.ft_epochs} --lambdas {params.lamb} --loss_func multitask_rmse --model rgcn -s 135" # """ diff --git a/river_dl/train_model.py b/river_dl/train_model_cli.py similarity index 90% rename from river_dl/train_model.py rename to river_dl/train_model_cli.py index fd8c86c..ba630aa 100644 --- a/river_dl/train_model.py +++ b/river_dl/train_model_cli.py @@ -1,3 +1,9 @@ +""" +This file provides a commandline interface (CLI) for the `train.train_model` +function. The commandline interface was originally provided to allow a command +to be sent to a slurm scheduler which was necessary to train the model using +GPUs. This has been tested on USGS's Tallgrass supercomputer. +""" import argparse from river_dl.train import train_model import river_dl.loss_functions as lf