Главная
/
Задания
/
Обучение AlphaZero для Quoridor 9x9 до 20 000 итераций на GPU-сервере. Есть готовый код на Python/PyTorch.
Видно всем
Обучение AlphaZero для Quoridor 9x9 до 20 000 итераций на GPU-сервере. Есть готовый код на Python/PyTorch.
Здравствуйте! У меня есть рабочий код AlphaZero для игры Quoridor на доске 9x9, написанный на Python с использованием PyTorch. Обучение запускается на моём домашнем компьютере, но идёт очень медленно — всего около 14 итераций в час. Мне нужно обучить модель до 20 000 итераций. Для этого я планирую арендовать мощный сервер с несколькими GPU, например, 8×RTX 6000 Pro или аналогичным, но мой текущий код не адаптирован под несколько видеокарт и при запуске на таком сервере будет использовать только одну из них, что сведёт на нет весь смысл аренды. Мне нужен специалист, который сможет адаптировать код для работы на нескольких GPU, используя DataParallel или DistributedDataParallel, либо предложить альтернативное решение, если текущая реализация плохо масштабируется. Обучение должно продолжиться с моих существующих чекпоинтов — у меня есть файлы best.pt и latest.pt, чтобы не начинать процесс заново. На выходе я хочу получить готовую обученную модель с 20 000 итераций, все промежуточные чекпоинты, логи обучения и краткий отчёт с итоговыми значениями. В отклике прошу описать ваш опыт с AlphaZero и распределённым обучением, какой подход вы предлагаете, сколько времени потребуется на настройку и запуск, а также какой сервер и провайдера вы рекомендуете для аренды. Буду рад обсудить детали.Требуемые навыки
Искусственный интеллект
Python
Pytorch
Разработка игр