update tf2

..._prediction_tools/data_extraction/extract_weatherbench.py

video_prediction_tools/main_scripts/main_train_models.py

...ediction_tools/main_scripts/main_visualize_postprocess.py

...n_tools/main_scripts/main_visualize_postprocess_gzprcp.py

...ediction_tools/model_modules/video_prediction/__init__.py

...tools/model_modules/video_prediction/datasets/__init__.py

..._tools/model_modules/video_prediction/datasets/dataset.py

...on_tools/model_modules/video_prediction/datasets/stats.py

...odel_modules/video_prediction/layers/BasicConvLSTMCell.py

..._tools/model_modules/video_prediction/layers/layer_def.py

...rediction_tools/model_modules/video_prediction/metrics.py

...del_modules/video_prediction/models/convLSTM_GAN_model.py

...s/model_modules/video_prediction/models/our_base_model.py

...modules/video_prediction/models/vanilla_convLSTM_model.py

...on_tools/model_modules/video_prediction/utils/tf_utils.py

video_prediction_tools/postprocess/plot_ambs_forecast.py