def test_transformer_train(test_dir): from src.bin import train config_path = "./unittests/configs/test_transformer.yaml" model_name = test_utils.get_model_name(config_path) saveto = os.path.join(test_dir, "save") log_path = os.path.join(test_dir, "log") valid_path = os.path.join(test_dir, "valid") train.run(model_name=model_name, config_path=config_path, saveto=saveto, log_path=log_path, valid_path=valid_path, debug=True)
def test_transformer_train(test_dir, use_gpu=False, multi_gpu=False): from src.bin import train config_path = "./unittests/configs/test_ocd_loss.yaml" model_name = test_utils.get_model_name(config_path) saveto = os.path.join(test_dir, "save") log_path = os.path.join(test_dir, "log") valid_path = os.path.join(test_dir, "valid") train.run(model_name=model_name, config_path=config_path, saveto=saveto, log_path=log_path, valid_path=valid_path, debug=True, use_gpu=use_gpu, multi_gpu=multi_gpu, task="ocd_old")
from src.bin import train if __name__ == '__main__': train.run()