We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent caeea6e commit a428e29Copy full SHA for a428e29
src/lightning/pytorch/loops/evaluation_loop.py
@@ -11,6 +11,7 @@
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14
+import math
15
import os
16
import shutil
17
import sys
@@ -268,7 +269,10 @@ def increment_progress_to_evaluation_end(self) -> None:
268
269
if self.skip:
270
return
271
self.reset()
- max_batch = int(max(self.max_batches))
272
+ max_batch = max(self.max_batches)
273
+ if isinstance(max_batch, float) and math.isinf(max_batch):
274
+ return
275
+ max_batch = int(max_batch)
276
if max_batch == -1:
277
278
self.batch_progress.increment_by(max_batch, True)
0 commit comments