Писал сам , но приходилось много много раз переписывать, и не к чему нормальному не пришел. Есть ли в pytorch свои функции или метрики (не знаю как правильно) что бы получать точность модели , loss доступна а вот accuracy приходится писать самому(безуспешно)