Remove flag from train

This commit is contained in:
thomashacker 2023-01-19 15:13:11 +01:00
parent d879e7280c
commit 400b5c7792

View File

@ -51,7 +51,6 @@ def train(
*,
use_gpu: int = -1,
overrides: Dict[str, Any] = util.SimpleFrozenDict(),
use_rehearse: bool = False,
):
config_path = util.ensure_path(config_path)
output_path = util.ensure_path(output_path)
@ -66,8 +65,6 @@ def train(
msg.good(f"Created output directory: {output_path}")
msg.info(f"Saving to output directory: {output_path}")
setup_gpu(use_gpu)
if use_rehearse:
msg.info(f"Using rehearsal updates")
with show_validation_error(config_path):
config = util.load_config(config_path, overrides=overrides, interpolate=False)
msg.divider("Initializing pipeline")