How to use the nncf.helpers.safe_thread_call function in nncf
To help you get started, we’ve selected a few nncf examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
opencv / openvino_training_extensions / pytorch_toolkit / nncf / examples / classification / main.py View on Github 
def create_cifar(config, dataset_config, is_train, transform):
create_cifar_fn = None
if dataset_config == 'cifar100':
create_cifar_fn = partial(CIFAR100, config.dataset_dir, train=is_train, transform=transform)
if dataset_config == 'cifar10':
create_cifar_fn = partial(CIFAR10, config.dataset_dir, train=is_train, transform=transform)
if create_cifar_fn:
return safe_thread_call(partial(create_cifar_fn, download=True), partial(create_cifar_fn, download=False))
return None
Popular nncf functions
- nncf.algo_selector.create_compression_algorithm
- nncf.compression_method_api.CompressionAlgorithm
- nncf.config.Config
- nncf.dynamic_graph.patch_torch_operators
- nncf.helpers.create_compressed_model
- nncf.helpers.load_state
- nncf.helpers.safe_thread_call
- nncf.layer_utils.COMPRESSION_MODULES
- nncf.layer_utils.COMPRESSION_MODULES.registry_dict.values
- nncf.layers.NNCF_MODULES_DICT.items
- nncf.quantization.layers.AsymmetricQuantizer
- nncf.quantization.layers.QuantizerConfig
- nncf.quantization.SymmetricQuantizer
- nncf.sparsity.functions.apply_binary_mask
- nncf.utils.dict_update
- nncf.utils.get_all_modules_by_type
- nncf.utils.get_node_name
- nncf.utils.in_scope_list
- nncf.utils.print_statistics