Skip to content
Snippets Groups Projects
Commit 6a2dfe63 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[scripts.train_analysis] Remove duplicate plots; Remove linear time plot (issue #20)

parent 37e0eeb1
No related branches found
No related tags found
1 merge request!24Implement fixes on evaluation (closes #20), and prepare for handling issue #60
......@@ -18,7 +18,6 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s")
def create_figures(
data: dict[str, tuple[list[int], list[float]]],
groups: list[str] = [
"total-execution-time-seconds",
"loss/*",
"learning-rate",
"memory-used-GB/cpu/*" "rss-GB/cpu/*",
......@@ -30,7 +29,6 @@ def create_figures(
"memory-percent/gpu/*",
"memory-used-GB/gpu/*",
"memory-free-GB/gpu/*",
"memory-free-GB/gpu/*",
"percent-usage/gpu/*",
],
) -> list:
......@@ -116,6 +114,7 @@ def create_figures(
)
@click.option(
"--logdir",
"-l",
help="Path to the directory containing the Tensorboard training logs",
required=True,
type=click.Path(dir_okay=True, exists=True, path_type=pathlib.Path),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment