Python add_datapath_and_model_args示例

示例#1

 def set_options(self, name, device):
     option = {}
     add_datapath_and_model_args(option)
     datapath = option.get("datapath")
     option['model_file'] = modelzoo_path(datapath, name)
     option["override"] = {
         "no_cuda": False if "cuda" in device else True,
     }
     return option

示例#2

文件： wow.py 项目： ibivibiv/openchat

def load_topics():
    opt = {}
    add_datapath_and_model_args(opt)
    build(opt)

    topics_path = os.path.join(
        opt['datapath'],
        'wizard_of_wikipedia',
        'topic_splits.json',
    )

    return sorted(json.load(open(topics_path, 'rb'))["train"])

示例#3

    def set_options(self, name, device):
        option = {
            "n_image_tokens": 1,
            "n_image_channels": 1,
            "image_fusion_type": "late",
        }

        add_datapath_and_model_args(option)
        datapath = option.get("datapath")
        option['model_file'] = modelzoo_path(datapath, name)
        option["override"] = {
            "no_cuda": False if "cuda" in device else True,
        }
        return option

示例#4

    def set_options(self, name, device):
        option = {}
        add_datapath_and_model_args(option)
        datapath = option.get("datapath")
        option['model_file'] = modelzoo_path(datapath, name)
        option["override"] = {
            "no_cuda": False if "cuda" in device else True,
        }

        if "cuda:" in device:
            option["override"]["gpu"] = int(device.split(":")[1])
        elif "cuda" in device:
            option["override"]["gpu"] = 0

        return option

示例#5

 def set_options(self, name, path, class_name, device):
     option = {
         "n_image_tokens": 1,
         "n_image_channels": 1,
         "image_fusion_type": "late",
     }
     add_datapath_and_model_args(option)
     datapath = option.get('datapath')
     option['model_file'] = modelzoo_path(datapath, name)
     option["override"] = {
         "no_cuda": False if "cuda" in device else True,
     }
     my_module = importlib.import_module(path)
     model_class = getattr(my_module, class_name)
     return option, model_class

示例#6

    def set_options(self, name, path, class_name, device):
        option = {
            "n_image_tokens": 1,
            "n_image_channels": 1,
            "image_fusion_type": "late",
            "image_features_dim": 2048,
            "image_encoder_num_layers": 1,
        }
        add_datapath_and_model_args(option)
        datapath = option.get('datapath')
        option['model_file'] = modelzoo_path(datapath, name)
        option["override"] = {
            "no_cuda": False if "cuda" in device else True,
        }

        if "cuda:" in device:
            option["override"]["gpu"] = int(device.split(":")[1])
        elif "cuda" in device:
            option["override"]["gpu"] = 0

        my_module = importlib.import_module(path)
        model_class = getattr(my_module, class_name)
        return option, model_class