[Fix] Fix BaseDataPreprocessor and BaseModel (#285)
* fix BaseDataPreprocessor * fix BaseDataPreprocessor * change device type to torch.device * change device type to torch.device * fix cpu method of base model
Showing
- mmengine/model/__init__.py 2 additions, 2 deletionsmmengine/model/__init__.py
- mmengine/model/base_model/base_model.py 16 additions, 6 deletionsmmengine/model/base_model/base_model.py
- mmengine/model/base_model/data_preprocessor.py 24 additions, 16 deletionsmmengine/model/base_model/data_preprocessor.py
- mmengine/model/utils.py 5 additions, 5 deletionsmmengine/model/utils.py
- tests/test_model/test_base_model/test_base_model.py 5 additions, 3 deletionstests/test_model/test_base_model/test_base_model.py
- tests/test_model/test_base_model/test_data_preprocessor.py 14 additions, 1 deletiontests/test_model/test_base_model/test_data_preprocessor.py
Loading
Please register or sign in to comment