tasks.py 514 Bytes
Newer Older
Yuan Liu's avatar
Yuan Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
from mmengine.config import read_base

with read_base():
    from .minigpt_4.minigpt_4_7b_mmbench import (minigpt_4_dataloader,
                                                 minigpt_4_evaluator,
                                                 minigpt_4_load_from,
                                                 minigpt_4_model)

models = [minigpt_4_model]
datasets = [minigpt_4_dataloader]
evaluators = [minigpt_4_evaluator]
load_froms = [minigpt_4_load_from]
13
14
15
num_gpus = 8
num_procs = 8
launcher = 'pytorch'