tasks.py 512 Bytes
Newer Older
Yuan Liu's avatar
Yuan Liu committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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]
num_gpus = 1
num_procs = 1
launcher = 'slurm'