from LMTrace.train import main

if __name__ == "__main__":
    args = {
        'train_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'valid_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'test_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        "model_name_or_path": "google/electra-small-discriminator",
        "model_arch": "electra_single",
        "output_dir": "./test_output/model",
        "do_train":True,
        "do_eval":True,
        # "do_predict": True,
        "overwrite_output_dir":True,
        "per_device_train_batch_size": 8,
        "num_train_epochs": 100,
        "logging_steps":10,
        "save_total_limit":10,
        "evaluation_strategy":"epoch",
        "run_name": 'test_run'
    }
    main(args)

0280cec42c727302bc3be5359e3ebbc25874ac7e
<mask token>

<mask token>
if __name__ == '__main__':
    args = {'train_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'valid_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'test_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'model_name_or_path': 'google/electra-small-discriminator',
        'model_arch': 'electra_single', 'output_dir': './test_output/model',
        'do_train': True, 'do_eval': True, 'overwrite_output_dir': True,
        'per_device_train_batch_size': 8, 'num_train_epochs': 100,
        'logging_steps': 10, 'save_total_limit': 10, 'evaluation_strategy':
        'epoch', 'run_name': 'test_run'}
    main(args)

from LMTrace.train import main
if __name__ == '__main__':
    args = {'train_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'valid_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'test_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'model_name_or_path': 'google/electra-small-discriminator',
        'model_arch': 'electra_single', 'output_dir': './test_output/model',
        'do_train': True, 'do_eval': True, 'overwrite_output_dir': True,
        'per_device_train_batch_size': 8, 'num_train_epochs': 100,
        'logging_steps': 10, 'save_total_limit': 10, 'evaluation_strategy':
        'epoch', 'run_name': 'test_run'}
    main(args)

from LMTrace.train import main

if __name__ == "__main__":
    args = {
        'train_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'valid_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        'test_dir': 'G:\\Projects\\ASE2021-NLTBert\\data\\test',
        "model_name_or_path": "google/electra-small-discriminator",
        "model_arch": "electra_single",
        "output_dir": "./test_output/model",
        "do_train":True,
        "do_eval":True,
        # "do_predict": True,
        "overwrite_output_dir":True,
        "per_device_train_batch_size": 8,
        "num_train_epochs": 100,
        "logging_steps":10,
        "save_total_limit":10,
        "evaluation_strategy":"epoch",
        "run_name": 'test_run'
    }
    main(args)
