Note
You are reading the documentation for MMClassification 0.x, which will soon be deprecated at the end of 2022. We recommend you upgrade to MMClassification 1.0 to enjoy fruitful new features and better performance brought by OpenMMLab 2.0. Check the installation tutorial, migration tutorial and changelog for more details.
mmcls.apis.single_gpu_test¶
- mmcls.apis.single_gpu_test(model, data_loader, show=False, out_dir=None, **show_kwargs)[source]¶
Test model with local single gpu.
This method tests model with a single gpu and supports showing results.
- Parameters
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.
- Returns
The prediction results.
- Return type