ml-workflow
stable
Contents:
workflow package
Setup project
Subpackages
workflow.functional package
workflow.ignite package
Subpackages
workflow.torch package
workflow.functional package
workflow.torch package
workflow.ignite package
Subpackages
workflow.ignite.decorators package
workflow.ignite.handlers package
workflow.ignite.handlers.learning_rate package
workflow.ignite.metrics package
ml-workflow
»
workflow package
»
workflow.ignite package
»
workflow.ignite.decorators package
Edit on GitHub
workflow.ignite.decorators package
¶
workflow.ignite.decorators.
train
(
model
,
optimizer
,
n_batches_per_step
=
1
)
[source]
¶
workflow.ignite.decorators.
evaluate
(
model
)
[source]
¶
workflow.ignite.decorators.
update_cpu_model
(
cpu_model
,
model
)
[source]
¶
workflow.ignite.decorators.
step
(
optimizer
,
n_batches_per_step
=
1
)
[source]
¶
workflow.ignite.decorators.
to_device
(
device
)
[source]
¶