Skip to content
Snippets Groups Projects
Commit 19988a54 authored by Tim Tobias Bauerle's avatar Tim Tobias Bauerle
Browse files

Removed printing optimizer state

parent be7a70f9
Branches
No related tags found
1 merge request!24Autoencoder, ATM and global optimizer
Hide whitespace changes
Inline Side-by-side
Showing
with 0 additions and 3 deletions
edml/core/server.py
+ 0
− 3
View file @ 19988a54
...@@ -92,10 +92,7 @@ class DeviceServer: ...@@ -92,10 +92,7 @@ class DeviceServer:
metrics = ModelMetricResultContainer() metrics = ModelMetricResultContainer()
diagnostic_metric_container = DiagnosticMetricResultContainer() diagnostic_metric_container = DiagnosticMetricResultContainer()
if optimizer_state is not None: if optimizer_state is not None:
print(f"apply optimizer state {optimizer_state}")
self._optimizer.load_state_dict(optimizer_state) self._optimizer.load_state_dict(optimizer_state)
else:
print("optimizer state is None")
if adaptive_learning_threshold is not None: if adaptive_learning_threshold is not None:
self.adaptive_learning_threshold = adaptive_learning_threshold self.adaptive_learning_threshold = adaptive_learning_threshold
for epoch in range(epochs): for epoch in range(epochs):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment