Skip to content
Snippets Groups Projects
Commit 07b66e90 authored by Michael Langguth's avatar Michael Langguth
Browse files

Minor fix in main_train_models.py.

parent aee36dd3
No related branches found
No related tags found
No related merge requests found
Pipeline #75870 passed
Stage: build
Stage: test
Stage: deploy
Show whitespace changes
Inline Side-by-side
video_prediction_tools/main_scripts/main_train_models.py
+ 1
− 1
View file @ 07b66e90
...@@ -378,7 +378,7 @@ class TrainModel(object): ...@@ -378,7 +378,7 @@ class TrainModel(object):
self.saver_loss = fetch_list[-1] self.saver_loss = fetch_list[-1]
self.saver_loss_name = "Total loss" self.saver_loss_name = "Total loss"
if self.video_model.__class__.__name__ == "ConvLstmGANVideoPredictionModel": if self.video_model.__class__.__name__ == "ConvLstmGANVideoPredictionModel":
fetch_list = fetch_list + [inputs, "total_loss", "inputs"] fetch_list = fetch_list + ["inputs", "total_loss"]
self.saver_loss = fetch_list[-1] self.saver_loss = fetch_list[-1]
self.saver_loss_name = "Total loss" self.saver_loss_name = "Total loss"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment