Akihiro Nitta
902774ab25
Specify `Trainer(benchmark=False)` in parity benchmarks ( #13182 )
...
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-06-01 08:04:16 -04:00
Kushashwa Ravi Shrimali
5df753404a
Bump version of black to 22.3.0 ( #12542 )
...
* bump black version to 22.3.0
2022-04-13 14:14:29 -04:00
Adrian Wälchli
dc4c3171fc
add parity test for sync batchnorm ( #12021 )
2022-02-26 03:51:57 +00:00
Aki Nitta
3c3ba39a06
Tests: Fail on FutureWarning ( #11541 )
2022-01-20 12:52:34 +00:00
Adrian Wälchli
c210e338ef
Update strategy import statements ( #11231 )
2021-12-23 08:26:28 +01:00
four4fish
cf5ef32f7b
Deprecate Trainer.training_type_plugin in favor of trainer.strategy ( #11141 )
...
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-12-22 02:11:43 +00:00
four4fish
4bfe5bda0f
Rename the DDPSpawnShardedPlugin to DDPSpawnShardeedStrategy ( #11210 )
...
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-12-22 00:27:36 +00:00
four4fish
b64dea9dc3
Rename `DDPPlugin` to `DDPStrategy` ( #11142 )
...
* Raname DDPPlugin to DDPStrategy
* Change ddp_plugin to ddp_strategy
* update changelog
* rename occurences in docs
* rename more occurrences
* fix line too long
* more fixes
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-12-21 08:55:51 +00:00
Carlos Mocholí
26977043bf
Add separate CI job for slow tests ( #10830 )
2021-12-01 19:58:18 +00:00
Carlos Mocholí
5788789f01
Move benchmarks into the test directory ( #10614 )
2021-11-19 03:07:33 +01:00