Update README.md

README.md

@@ -27,7 +27,10 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 
 ## Используемая loss функция
 
-Для обучения модели была использована loss функция `nn.KLDivLoss`, которая применима для задач, где модель генерирует вероятностные распределения.
+Для обучения модели была использована loss функция `nn.KLDivLoss'.
+Дообучал только добавленные слои.
+
+Также пробавал обучить всю модель с MSE loss. При этом модель запомнила средние вероятности и не меняла предсказания от изменения входных данных.
 
 ## Датасет