From 36e2961921c3957924b796e738012dbece46c477 Mon Sep 17 00:00:00 2001 From: Jerome Lecoq Date: Mon, 6 Dec 2021 14:56:42 -0800 Subject: [PATCH] Linting --- deepinterpolation/cli/schemas.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deepinterpolation/cli/schemas.py b/deepinterpolation/cli/schemas.py index 1013cd59..bfd4b1e6 100644 --- a/deepinterpolation/cli/schemas.py +++ b/deepinterpolation/cli/schemas.py @@ -488,6 +488,7 @@ class TrainingSchema(argschema.schemas.DefaultSchema): training becomes limited either by RAM or CPU usage.", ) + class FineTuningSchema(argschema.schemas.DefaultSchema): name = argschema.fields.String( required=False, @@ -592,7 +593,7 @@ class FineTuningSchema(argschema.schemas.DefaultSchema): description="Period in number of epochs to periodically save model \ checkpoints.", ) - + use_multiprocessing = argschema.fields.Bool( required=False, default=True, @@ -613,6 +614,7 @@ class FineTuningSchema(argschema.schemas.DefaultSchema): training becomes limited either by RAM or CPU usage.", ) + class NetworkSchema(argschema.schemas.DefaultSchema): name = argschema.fields.String( required=True,