- 26 Jul, 2021 1 commit
-
-
Yuting Jiang authored
**Description** Fix bug of VGG models failed on A100 GPU with batch_size=128. **Major Revision** - Add synchronize before delete self.model in _postprocess of superbench/benchmarks/model_benchmarks/pytorch_base.py **Minor Revision** - Add if self._gpu_available check - Reorder del
-
- 28 Jun, 2021 1 commit
-
-
guoshzhao authored
* replace torch.optim.AdamW with transformers.AdamW.
-
- 07 Jun, 2021 1 commit
-
-
guoshzhao authored
* Clean up the cache.
-
- 19 May, 2021 1 commit
-
-
Yuting Jiang authored
-
- 12 Apr, 2021 1 commit
-
-
guoshzhao authored
Co-authored-by:Guoshuai Zhao <guzhao@microsoft.com>
-
- 15 Mar, 2021 1 commit
-
-
guoshzhao authored
Co-authored-by:
-
- 09 Mar, 2021 2 commits
-
-
guoshzhao authored
* add flag to disable GPU. * fix spelling * fix unittest. * address comments. Co-authored-by: -
guoshzhao authored
Co-authored-by:
-
- 08 Mar, 2021 1 commit
-
-
guoshzhao authored
* add pytorch base class * address comments Co-authored-by:
-
