Skip to content

GitLab

"test/srt/rl/test_verl_engine_4_gpu.py" did not exist on "0194948fd96802ea41d6baa3673e9d42cfa5dc8c"
Switch branch/tag
Find file BlameHistoryPermalink
  • Yuting Jiang's avatar
    fix bug of cnn failure (#134) · 76a611f7
    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
    76a611f7
pytorch_base.py 7.54 KB