Skip to content
Snippets Groups Projects
Commit 929614e0 authored by Bing Gong's avatar Bing Gong
Browse files

Update video_prediction_tools/main_scripts/main_train_models.py

parent 08913a21
No related branches found
No related tags found
No related merge requests found
Pipeline #125930 failed
Stage: build
Stage: test
Stage: deploy
Show whitespace changes
Inline Side-by-side
video_prediction_tools/main_scripts/main_train_models.py
+ 0
− 1
View file @ 929614e0
...@@ -418,7 +418,6 @@ class TrainModel(object): ...@@ -418,7 +418,6 @@ class TrainModel(object):
if self.video_model.__class__.__name__ == "VanillaConvLstmVideoPredictionModel": if self.video_model.__class__.__name__ == "VanillaConvLstmVideoPredictionModel":
fetch_list = fetch_list + ["inputs"] fetch_list = fetch_list + ["inputs"]
self.saver_loss = fetch_list[-1] self.saver_loss = fetch_list[-1]
<<<<<<< HEAD
self.saver_loss_name = "Total loss" self.saver_loss_name = "Total loss"
if self.video_model.__class__.__name__ == "WeatherBenchModel": if self.video_model.__class__.__name__ == "WeatherBenchModel":
fetch_list = fetch_list + ["total_loss"] fetch_list = fetch_list + ["total_loss"]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment