Spaces:
Running
Running
File size: 2,914 Bytes
f51c1fd a0a9f5d f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd 8b87ced e0ca321 f51c1fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
"""
File: oceanai_init.py
Author: Elena Ryumina and Dmitry Ryumin
Description: OceanAI initialization.
License: MIT License
"""
from oceanai.modules.lab.build import Run
def oceanai_initialization():
out = False
# Создание экземпляра класса
_b5 = Run(lang="en", metadata=out)
# Настройка ядра
_b5.path_to_save_ = "./models" # Директория для сохранения файла
_b5.chunk_size_ = 2000000 # Размер загрузки файла из сети за 1 шаг
corpus = "fi"
disk = "googledisk"
# Формирование аудиомоделей
_ = _b5.load_audio_model_hc(out=out)
_ = _b5.load_audio_model_nn(out=out)
# Загрузка весов аудиомоделей
url = _b5.weights_for_big5_["audio"][corpus]["hc"][disk]
_ = _b5.load_audio_model_weights_hc(url=url, out=out)
url = _b5.weights_for_big5_["audio"][corpus]["nn"][disk]
_ = _b5.load_audio_model_weights_nn(url=url, out=out)
# Формирование видеомоделей
_ = _b5.load_video_model_hc(lang="en", out=out)
_ = _b5.load_video_model_deep_fe(out=out)
_ = _b5.load_video_model_nn(out=out)
# Загрузка весов видеомоделей
url = _b5.weights_for_big5_["video"][corpus]["hc"][disk]
_ = _b5.load_video_model_weights_hc(url=url, out=out)
url = _b5.weights_for_big5_["video"][corpus]["fe"][disk]
_ = _b5.load_video_model_weights_deep_fe(url=url, out=out)
url = _b5.weights_for_big5_["video"][corpus]["nn"][disk]
_ = _b5.load_video_model_weights_nn(url=url, out=out)
# Загрузка словаря с экспертными признаками (текстовая модальность)
_ = _b5.load_text_features(out=out)
# Формирование текстовых моделей
_ = _b5.setup_translation_model() # только для русского языка
_ = _b5.setup_bert_encoder(force_reload=False, out=out)
_ = _b5.load_text_model_hc(corpus=corpus, out=out)
_ = _b5.load_text_model_nn(corpus=corpus, out=out)
# Загрузка весов текстовых моделей
url = _b5.weights_for_big5_["text"][corpus]["hc"][disk]
_ = _b5.load_text_model_weights_hc(url=url, out=out)
url = _b5.weights_for_big5_["text"][corpus]["nn"][disk]
_ = _b5.load_text_model_weights_nn(url=url, out=out)
# Формирование модели для мультимодального объединения информации
_ = _b5.load_avt_model_b5(out=out)
# Загрузка весов модели для мультимодального объединения информации
url = _b5.weights_for_big5_["avt"][corpus]["b5"][disk]
_ = _b5.load_avt_model_weights_b5(url=url, out=out)
return _b5
b5 = oceanai_initialization()
|