search

pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.5k stars 608 forks source link

adjust tpu timeout #3246

Closed leej3 closed 2 months ago

leej3 commented 2 months ago

moves collection time test modification into the main hook for modifying tests