Skip to content

Commit

Permalink
Linting
Browse files Browse the repository at this point in the history
  • Loading branch information
jeromelecoq committed Dec 6, 2021
1 parent 64161af commit 36e2961
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion deepinterpolation/cli/schemas.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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,
Expand All @@ -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,
Expand Down

0 comments on commit 36e2961

Please sign in to comment.