# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     language: python
#     name: python3
# ---

class TrainingConfig(object):

    # set training config

    #BERT training config
    numTrainEpochs = 1
    learningRate = 2e-5
    batchSize = 8
    warmUpProportion = 0.1

    #LSTM training config
    batch_size_train = 2
    batches_per_epoch_train = 91
    batch_size_val = 3
    batches_per_epoch_val = 9
    batch_size_test = 1
    batches_per_epoch_test = 19 

StarcoderdataJupyterScriptsDedupFiltered