ElenaRyumina commited on
Commit
e0ca321
·
1 Parent(s): 87a92fb

Update oceanai_init.py

Browse files
Files changed (1) hide show
  1. app/oceanai_init.py +9 -9
app/oceanai_init.py CHANGED
@@ -26,10 +26,10 @@ def oceanai_initialization():
26
 
27
  # Загрузка весов аудиомоделей
28
  url = _b5.weights_for_big5_["audio"][corpus]["hc"]["sberdisk"]
29
- _ = _b5.load_audio_model_weights_hc(url=url, out=out, force_reload=False)
30
 
31
  url = _b5.weights_for_big5_["audio"][corpus]["nn"]["sberdisk"]
32
- _ = _b5.load_audio_model_weights_nn(url=url, out=out, force_reload=False)
33
 
34
  # Формирование видеомоделей
35
  _ = _b5.load_video_model_hc(lang="en", out=out)
@@ -38,16 +38,16 @@ def oceanai_initialization():
38
 
39
  # Загрузка весов видеомоделей
40
  url = _b5.weights_for_big5_["video"][corpus]["hc"]["sberdisk"]
41
- _ = _b5.load_video_model_weights_hc(url=url, out=out, force_reload=False)
42
 
43
  url = _b5.weights_for_big5_["video"][corpus]["fe"]["sberdisk"]
44
- _ = _b5.load_video_model_weights_deep_fe(url=url, out=out, force_reload=False)
45
 
46
  url = _b5.weights_for_big5_["video"][corpus]["nn"]["sberdisk"]
47
- _ = _b5.load_video_model_weights_nn(url=url, out=out, force_reload=False)
48
 
49
  # Загрузка словаря с экспертными признаками (текстовая модальность)
50
- _ = _b5.load_text_features(out=out, force_reload=False)
51
 
52
  # Формирование текстовых моделей
53
  _ = _b5.setup_translation_model() # только для русского языка
@@ -57,17 +57,17 @@ def oceanai_initialization():
57
 
58
  # Загрузка весов текстовых моделей
59
  url = _b5.weights_for_big5_["text"][corpus]["hc"]["sberdisk"]
60
- _ = _b5.load_text_model_weights_hc(url=url, out=out, force_reload=False)
61
 
62
  url = _b5.weights_for_big5_["text"][corpus]["nn"]["sberdisk"]
63
- _ = _b5.load_text_model_weights_nn(url=url, out=out, force_reload=False)
64
 
65
  # Формирование модели для мультимодального объединения информации
66
  _ = _b5.load_avt_model_b5(out=out)
67
 
68
  # Загрузка весов модели для мультимодального объединения информации
69
  url = _b5.weights_for_big5_["avt"][corpus]["b5"]["sberdisk"]
70
- _ = _b5.load_avt_model_weights_b5(url=url, out=out, force_reload=False)
71
 
72
  return _b5
73
 
 
26
 
27
  # Загрузка весов аудиомоделей
28
  url = _b5.weights_for_big5_["audio"][corpus]["hc"]["sberdisk"]
29
+ _ = _b5.load_audio_model_weights_hc(url=url, out=out)
30
 
31
  url = _b5.weights_for_big5_["audio"][corpus]["nn"]["sberdisk"]
32
+ _ = _b5.load_audio_model_weights_nn(url=url, out=out)
33
 
34
  # Формирование видеомоделей
35
  _ = _b5.load_video_model_hc(lang="en", out=out)
 
38
 
39
  # Загрузка весов видеомоделей
40
  url = _b5.weights_for_big5_["video"][corpus]["hc"]["sberdisk"]
41
+ _ = _b5.load_video_model_weights_hc(url=url, out=out)
42
 
43
  url = _b5.weights_for_big5_["video"][corpus]["fe"]["sberdisk"]
44
+ _ = _b5.load_video_model_weights_deep_fe(url=url, out=out)
45
 
46
  url = _b5.weights_for_big5_["video"][corpus]["nn"]["sberdisk"]
47
+ _ = _b5.load_video_model_weights_nn(url=url, out=out)
48
 
49
  # Загрузка словаря с экспертными признаками (текстовая модальность)
50
+ _ = _b5.load_text_features(out=out)
51
 
52
  # Формирование текстовых моделей
53
  _ = _b5.setup_translation_model() # только для русского языка
 
57
 
58
  # Загрузка весов текстовых моделей
59
  url = _b5.weights_for_big5_["text"][corpus]["hc"]["sberdisk"]
60
+ _ = _b5.load_text_model_weights_hc(url=url, out=out)
61
 
62
  url = _b5.weights_for_big5_["text"][corpus]["nn"]["sberdisk"]
63
+ _ = _b5.load_text_model_weights_nn(url=url, out=out)
64
 
65
  # Формирование модели для мультимодального объединения информации
66
  _ = _b5.load_avt_model_b5(out=out)
67
 
68
  # Загрузка весов модели для мультимодального объединения информации
69
  url = _b5.weights_for_big5_["avt"][corpus]["b5"]["sberdisk"]
70
+ _ = _b5.load_avt_model_weights_b5(url=url, out=out)
71
 
72
  return _b5
73