"include/vscode:/vscode.git/clone" did not exist on "5b5d62f19019a398167df1f1b59279e049bf24ce"
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