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.
16 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 'main'
${ noResults }
Go to file
Vladimir 96d18b4873 maybe more readable code 6 months ago
checkpoints update 7 months ago
data update 7 months ago
logs added padding mask 6 months ago
src maybe more readable code 6 months ago
.dockerignore IC 7 months ago
.gitattributes IC 7 months ago
.gitignore IC 7 months ago
Dockerfile IC 7 months ago
build.sh IC 7 months ago
readme.md dp and ddp update 6 months ago
start.sh IC 7 months ago

readme.md
Unescape Escape

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

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

# single gpu версия
CUDA_VISIBLE_DEVICES=0 python src/bert_training.py experiment_name
# data parallel версия 
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python src/bert_training_dp.py experiment_name
# 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 experiment_name

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

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