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
No related branches found
No related tags found
1 merge request!24Autoencoder, ATM and global optimizer
...@@ -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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment