diff --git a/mala/network/predictor.py b/mala/network/predictor.py index a313e4fcb..0e1c6e484 100644 --- a/mala/network/predictor.py +++ b/mala/network/predictor.py @@ -191,7 +191,6 @@ def _forward_snap_descriptors( self, snap_descriptors, local_data_size=None ): """Forward a scaled tensor of descriptors through the NN.""" - # Ensure the Network is on the correct device. # This line is necessary because GPU acceleration may have been # activated AFTER loading a model.