reg_log.py

from parsing import parse_train_args
from cv import cv
from utils import create_logger

if __name__ == '__main__':
    args = parse_train_args()

    args.num_folds=1
    args.epochs =1
    args.ensemble_size=1
    args.batch_size=128

    args.activation='ReLU'
    args.depth=4
    args.dropout=0.25
    args.ffn_num_layers=2
    args.hidden_size=384
    args.diff_depth_weights=False

    args.attention=True
    args.sumstyle=True
    args.seed=3032


    args.data_path, args.cols_to_read='data_RE2/LogP_moleculenet.csv', [x for x in range(2)]
    args.save_dir='save_test'
    args.gpuUSE=True
    args.gpu=2
    args.dataset_type, args.metric='regression','rmse'
    args.scale='normalization'
    args.tmp_data_dir='./data_RE2/tmp/'
    args.split_type='random'



    args.diff_depth_weights=False
    args.layers_per_message=1
    args.log_dir=None

    print(args)
    logger = create_logger(name='train_crossValidate', save_dir=args.save_dir, quiet=args.quiet)
    cv(args, logger)