fix bug of cnn failure (#134)
**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
Showing
Please register or sign in to comment