fine_tune

Attributes

`pos_type`
`add_feature_mask`
`model_name`
`output_dir`
`args`
`local`
`train_messages_len`
`test_messages_len`
`inference_batch_size`
`dataset_types`
`best_checkpoint`
`dataset`

Module Contents

fine_tune.pos_type

fine_tune.add_feature_mask = True

fine_tune.model_name = 'meta-llama/Llama-3.2-1B-Instruct'

fine_tune.output_dir = './model/'

fine_tune.args

fine_tune.local = False

fine_tune.train_messages_len = 6000

fine_tune.test_messages_len = 1319

fine_tune.inference_batch_size = 1

fine_tune.dataset_types = ['base']

fine_tune.best_checkpoint = [None]

fine_tune.dataset