备注
您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。
mmcls.apis.single_gpu_test¶
- mmcls.apis.single_gpu_test(model, data_loader, show=False, out_dir=None, **show_kwargs)[源代码]¶
Test model with local single gpu.
This method tests model with a single gpu and supports showing results.
- 参数
model (
torch.nn.Module) – Model to be tested.data_loader (
torch.utils.data.DataLoader) – Pytorch data loader.show (bool) – Whether to show the test results. Defaults to False.
out_dir (str) – The output directory of result plots of all samples. Defaults to None, which means not to write output files.
**show_kwargs – Any other keyword arguments for showing results.
- 返回
The prediction results.
- 返回类型