LongFinBERT-base / train_v1a_0803_1144_seed_1.log
minhtriphan's picture
Upload 6 files
781473a
raw
history blame
1.46 kB
11:44:11 {'seed': 1, 'ver': 'v1a', 'use_log': True, 'use_tqdm': True, 'debug': False, 'tokenizer': BertTokenizerFast(name_or_path='tokenizer', vocab_size=52000, model_max_length=512, is_fast=True, padding_side='right', truncation_side='right', special_tokens={'unk_token': '[UNK]', 'sep_token': '[SEP]', 'pad_token': '[PAD]', 'cls_token': '[CLS]', 'mask_token': '[MASK]'}, clean_up_tokenization_spaces=True), 'config': <custom_config.LongBERTConfig object at 0x7f2b78440110>, 'max_len': 45000, 'train_one_part': False, 'gradient_accumulation_steps': 2, 'apex': True, 'device': device(type='cuda', index=1), 'nepochs': 4, 'batch_size': 2, 'num_workers': 128, 'lr': 2e-05, 'weight_decay': 0.01, 'encoder_lr': 2e-05, 'decoder_lr': 0.001, 'min_lr': 1e-06, 'eps': 1e-06, 'betas': (0.9, 0.999), 'scheduler_type': 'cosine', 'num_cycles': 0.5, 'num_warmup_steps': 0.0, 'train_data_dir': 'data/train', 'valid_data_dir': 'data/valid', 'test_data_dir': '.', 'output_dir': 'model/v1/a'}
11:44:11 Preparing training materials...
11:44:11 Preparing the model...
11:44:13 Preparing the dataloaders...
05:03:38 Epoch: [1] - Train/Valid Loss: 5.3361/4.9918
05:03:38 Saving the model to model/v1/a
22:26:24 Epoch: [2] - Train/Valid Loss: 4.9233/4.8049
22:26:24 Saving the model to model/v1/a
15:48:34 Epoch: [3] - Train/Valid Loss: 4.7910/4.7136
15:48:34 Saving the model to model/v1/a
09:11:11 Epoch: [4] - Train/Valid Loss: 4.7136/4.6469
09:11:11 Saving the model to model/v1/a