Browse Source

commit code

master
行嗔 3 years ago
parent
commit
ca72f5329c
5 changed files with 7 additions and 0 deletions
  1. +2
    -0
      modelscope/metrics/accuracy_metric.py
  2. +0
    -0
      modelscope/models/multi_modal/ofa/adaptor/__init__.py
  3. +1
    -0
      modelscope/models/multi_modal/ofa/modeling_ofa.py
  4. +2
    -0
      modelscope/trainers/multi_modal/ofa/__init__.py
  5. +2
    -0
      modelscope/trainers/multi_modal/ofa/ofa_trainer.py

+ 2
- 0
modelscope/metrics/accuracy_metric.py View File

@@ -1,3 +1,5 @@
# Copyright (c) Alibaba, Inc. and its affiliates.

from typing import Dict

import numpy as np


+ 0
- 0
modelscope/models/multi_modal/ofa/adaptor/__init__.py View File


+ 1
- 0
modelscope/models/multi_modal/ofa/modeling_ofa.py View File

@@ -54,6 +54,7 @@ OFA_PRETRAINED_MODEL_ARCHIVE_LIST = [
'ofa-medium',
'ofa-base',
'ofa-large',
'ofa-huge',
]

try:


+ 2
- 0
modelscope/trainers/multi_modal/ofa/__init__.py View File

@@ -1 +1,3 @@
# Copyright (c) Alibaba, Inc. and its affiliates.

from .ofa_trainer import OFATrainer

+ 2
- 0
modelscope/trainers/multi_modal/ofa/ofa_trainer.py View File

@@ -1,3 +1,5 @@
# Copyright (c) Alibaba, Inc. and its affiliates.

import math
import os
from functools import partial


Loading…
Cancel
Save