mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 10:30:45 +00:00
19 lines
538 B
Python
19 lines
538 B
Python
"""
|
|
Copyright (c) 2022, salesforce.com, inc.
|
|
All rights reserved.
|
|
SPDX-License-Identifier: BSD-3-Clause
|
|
For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause
|
|
"""
|
|
|
|
from minigpt4.common.registry import registry
|
|
from minigpt4.tasks.base_task import BaseTask
|
|
|
|
|
|
@registry.register_task("image_text_pretrain")
|
|
class ImageTextPretrainTask(BaseTask):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def evaluation(self, model, data_loader, cuda_enabled=True):
|
|
pass
|