|
a |
|
b/minigpt4/tasks/image_text_pretrain.py |
|
|
1 |
""" |
|
|
2 |
Copyright (c) 2022, salesforce.com, inc. |
|
|
3 |
All rights reserved. |
|
|
4 |
SPDX-License-Identifier: BSD-3-Clause |
|
|
5 |
For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause |
|
|
6 |
""" |
|
|
7 |
|
|
|
8 |
from minigpt4.common.registry import registry |
|
|
9 |
from minigpt4.tasks.base_task import BaseTask |
|
|
10 |
|
|
|
11 |
|
|
|
12 |
@registry.register_task("image_text_pretrain") |
|
|
13 |
class ImageTextPretrainTask(BaseTask): |
|
|
14 |
def __init__(self): |
|
|
15 |
super().__init__() |
|
|
16 |
|
|
|
17 |
def evaluation(self, model, data_loader, cuda_enabled=True): |
|
|
18 |
pass |