Вопросы организаторам

06.03.2019, 18:54:49
Andrey Selukov
Cтавлю pip install mxnet-cu92, а все равно программа
https://github.com/MailRuChamps/snahackathon/blob/master/2019/Images.ipynb
CPU гоняет, как включить GPU, код программы нужно изменять или автоматом gluon-cv GPU подхватывает?

P.S. CUDA, NVIDIA cuDNN все стоит, все работает.

меняем две строки в коде
"net = model_zoo.get_model('yolo3_darknet53_coco', pretrained=True)"
на
"net = model_zoo.get_model('yolo3_darknet53_coco', pretrained=True, ctx=mxnet.gpu(0))"

"class_IDs, scores_v, _ = net(x)",
на
"class_IDs, scores_v, _ = net(x.as_in_context(mxnet.gpu(0)))"

и добавляем >export MXNET_CUDNN_AUTOTUNE_DEFAULT=0

получаем 4 часа на буке вместо 9. У кого видео карта веселей, еще меньше часов.
Отредактировано 06.03.2019, 19:02:11
06.03.2019, 18:56:41
вот блин, ошибся, как удалить сообщение
Отредактировано 06.03.2019, 19:01:49
07.03.2019, 15:59:23
Дмитрий Бугайченко
Иван Брагин
Ура, наконец-то вы вернулись, отличные задачки.
Вопрос по метрике.
У всех пользователей в тестовом наборе есть хотя бы один объект каждого класса?
Если нет, подскажите, чему равно ROC AUC для предсказаний, GT которых имеют только один класс? а то scikit например ругается в таком случае.

Привет старым знакомым :)!

Да, все так - в тесте только пользователи по которым есть и хотя бы один позитив, и хотя бы один негатив.

Правильно понимаю, что под позитив подразумевается Liked, а под негатив - Disliked? Примеры только с Clicked сюда не относятся? Будет ли в тесте юзер, который, например, только кликал на записи?
Отредактировано 07.03.2019, 15:59:54


Зарегистрируйтесь или войдите, чтобы оставить сообщение.