def entrance(save_model_path='/mnt/chjinche/test_data/detection/init_model', model_type='fasterrcnn_resnet50_fpn', pretrained=True): model_config = { 'model_class': 'FasterRCNN', 'model_type': model_type, 'pretrained': pretrained } logger.info('Dump untrained model.') logger.info(f'Model config: {model_config}.') dumper = pickle_dumper(model_config, 'model_config.pkl') save_model_to_directory(save_model_path, dumper) logger.info('Finished.')
def entrance( save_model_path='/mnt/chjinche/projects/saved_untrained_model_resnet', model_type='resnext101_32x8d', pretrained=True): model_config = { 'model_class': 'ResNet', 'model_type': model_type, 'pretrained': pretrained } logger.info('Dump untrained model.') logger.info(f'Model config: {model_config}.') dumper = pickle_dumper(model_config, 'model_config.pkl') save_model_to_directory(save_model_path, dumper) logger.info('Finished.')
def entrance(save_model_path='../init_model', model_type='densenet201', pretrained=True, memory_efficient=False): model_config = { 'model_class': 'DenseNet', 'model_type': model_type, 'pretrained': pretrained, 'memory_efficient': memory_efficient } logger.info('Dump untrained model.') logger.info(f'Model config: {model_config}.') dumper = pickle_dumper(model_config, 'model_config.pkl') save_model_to_directory(save_model_path, dumper) logger.info('Finished.')