vlpr
/
alfabank_challenge
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 Commits
1 Branch
0 Tags
674 MiB
Python 96.1%
Dockerfile 2.6%
Shell 1.3%
 
 
 
main
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '8553c61dc9'
${ noResults }
Go to file
Vladimir 8553c61dc9 update 11 months ago
checkpoints update 12 months ago
data IC 12 months ago
runs update 12 months ago
src update 11 months ago
.dockerignore IC 12 months ago
.gitattributes IC 12 months ago
.gitignore IC 12 months ago
Dockerfile IC 12 months ago
build.sh IC 12 months ago
readme.md update 12 months ago
start.sh IC 12 months ago

readme.md
Unescape Escape

Скрипты для обучения bert предсказанию кредитного дефолта по данным кредитных историй

https://ods.ai/competitions/dl-fintech-bki

# single gpu версия
python src/bert_training.py fold3_18l_dyt_04_04_3750
# data parallel версия 
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python src/bert_training_dp.py fold3_18l_dyt_04_04_3750
# distributed data parallel версия 
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --standalone --nproc-per-node=8 src/bert_training_ddp.py fold3_18l_dyt_04_04_3750

Логирование ведётся в tensorboard в папку ./runs/. В папку с логами при запуске копируется текущая версия скрипта. Чекпоинты моделей сохраняются в папку ./checkpoints/.

Разбиение на обучающую и тестовую выборки осуществляется скриптом train_test_split.py.