add notifications to files that have changed a lot

compared to their counterparts in the original repo

bin/main.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

src/ml_downscaling_emulator/score_sde_pytorch_hja22/losses.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for sampling
 
 """All functions related to loss computation and optimization.
 """

src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/cncsnpp.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Significant modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling
 
 # pylint: skip-file
 

src/ml_downscaling_emulator/score_sde_pytorch_hja22/models/utils.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling.
 
 """All functions and modules related to model definition.
 """

src/ml_downscaling_emulator/score_sde_pytorch_hja22/run_lib.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Significant modifications to the original work have been made by Henry Addison
+# to allow for conditional modelling, location-specific parameters,
+# removal of tensorflow dependency, tracking for training via Weights and Biases
+# and MLFlow, and iterating by epoch using PyTorch DataLoaders
 
 # pylint: skip-file
 """Training and evaluation for score-based generative models. """

src/ml_downscaling_emulator/score_sde_pytorch_hja22/sampling.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright 2020 The Google Research Authors.
+# Modifications copyright 2024 Henry Addison
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,6 +13,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
+# Modifications to the original work have been made by Henry Addison
+# to allow for sampling
 
 # pylint: skip-file
 # pytype: skip-file