mirrors/mmfewshot
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmfewshot.git synced 2025-06-03 14:49:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Enhancement] Set Num thread to 0 (#36)

Browse Source 
* [Docs] update batch size

* [Enhancement] Set Num thread to 0

* set omp mkl threads numbers

* set cls omp mkl threads numbers

* add comments
This commit is contained in:
Linyiqi 2022-01-27 20:19:14 +08:00 committed by GitHub
parent 555a348f97
commit 66d92a5abb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.cfg
View File

@ -3,7 +3,7 @@ line_length = 79
multi_line_output = 0 multi_line_output = 0
known_standard_library = setuptools known_standard_library = setuptools
known_first_party = mmfewshot known_first_party = mmfewshot
known_third_party = mmcls,mmcv,mmdet,numpy,pytest,pytorch_sphinx_theme,terminaltables,torch,typing_extensions known_third_party = cv2,mmcls,mmcv,mmdet,numpy,pytest,pytorch_sphinx_theme,terminaltables,torch,typing_extensions
no_lines_before = STDLIB,LOCALFOLDER no_lines_before = STDLIB,LOCALFOLDER
default_section = THIRDPARTY default_section = THIRDPARTY

6
tools/classification/train.py
View File

@ -5,6 +5,7 @@ import os
import os.path as osp import os.path as osp
import time import time
import cv2
import mmcv import mmcv
import torch import torch
from mmcls import __version__ from mmcls import __version__
@ -17,6 +18,11 @@ from mmfewshot.classification.apis import train_model
from mmfewshot.classification.datasets import build_dataset from mmfewshot.classification.datasets import build_dataset
from mmfewshot.utils import get_root_logger from mmfewshot.utils import get_root_logger
# disable multithreading to avoid system being overloaded
cv2.setNumThreads(0)
os.environ['OMP_NUM_THREADS'] = '1'
os.environ['MKL_NUM_THREADS'] = '1'
def parse_args(): def parse_args():
parser = argparse.ArgumentParser(description='Train a model') parser = argparse.ArgumentParser(description='Train a model')

6
tools/detection/train.py
View File

@ -6,6 +6,7 @@ import os.path as osp
import time import time
import warnings import warnings
import cv2
import mmcv import mmcv
import torch import torch
from mmcv import Config, DictAction from mmcv import Config, DictAction
@ -20,6 +21,11 @@ from mmfewshot.detection.datasets import build_dataset
from mmfewshot.detection.models import build_detector from mmfewshot.detection.models import build_detector
from mmfewshot.utils import get_root_logger from mmfewshot.utils import get_root_logger
# disable multithreading to avoid system being overloaded
cv2.setNumThreads(0)
os.environ['OMP_NUM_THREADS'] = '1'
os.environ['MKL_NUM_THREADS'] = '1'
def parse_args(): def parse_args():
parser = argparse.ArgumentParser(description='Train a FewShot model') parser = argparse.ArgumentParser(description='Train a FewShot model')