Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ tests/tests_pytorch/loops/test_prediction_loop.py August 13, 2025
J
chore: bump PyTorch version in dependencies & CI (#21043)
Jirka Borovec committed
January 7, 2025
P
[pre-commit.ci] pre-commit suggestions (#20532)
pre-commit-ci[bot] committed
March 17, 2024
F
test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/loops` (#19646)
Felix Hirwa Nshuti committed
February 23, 2024
A
Update tests for PyTorch 2.2.1 (#19521)
awaelchli committed
February 15, 2024
J
precommit: ruff-format (#19434)
Jirka Borovec committed
January 26, 2024
J
CI: enable testing with coming PT 2.2 (#19289)
Jirka Borovec committed
September 26, 2023
J
ruff: replace isort with ruff +TPU (#17684)
Jirka Borovec committed
September 11, 2023
Q
Make the batch_idx argument optional in all step methods (#18512)
qunhong zneg committed
September 5, 2023
C
Support multiple dataloaders with `dataloader_iter` (#18390)
Carlos Mocholí committed
May 5, 2023
J
ruff: PT some more fixes (#17569)
Jirka Borovec committed
February 27, 2023
C
Check for dataloader_idx presence in the hooks (#16837)
Carlos Mocholí committed
February 23, 2023
C
Fabric: do `set_epoch` for `batch_sampler.sampler` (#16841)
Carlos Mocholí committed
February 22, 2023
C
Consume the prediction batch indices iteratively (#16826)
Carlos Mocholí committed
February 20, 2023
C
Remove `Trainer(multiple_trainloader_mode)` in favor of `CombinedLoader(mode)` (#16800)
Carlos Mocholí committed
February 17, 2023
C
Sequential `CombinedLoader` to flatten the eval and predict loops (#16726)
Carlos Mocholí committed
February 16, 2023
C
Fix `set_epoch` not getting called for prediction dataloaders (#16785)
Carlos Mocholí committed
February 7, 2023
C
Remove outputs from `on_predict_epoch_end` (#16655)
Carlos Mocholí committed