from data.load_data import SupportedDatasets
from train.pretrain_force import pretrain_force

'''
scp 1500011335@115.27.161.31:yangshuwen/GeometryEncoding/train/force/* train/force
scp 1500011335@115.27.161.31:yangshuwen/GeometryEncoding/log/force/* log/force
'''

if __name__ == '__main__':
    pretrain_force(
        dataset_name=SupportedDatasets.QM7,
        token='force-qm7',
        use_tqdm=False,
        use_cuda=True
    )
    # pretrain_force(
    #     use_tqdm=False,
    #     use_cuda=True
    # )
