trainer/trainer.py

from trainer.utils import *
from configs.configs import root,mae_config

def main():
    try:
        print(f"Training mae")
        trainer=MAETrainer(mae_config)
        trainer.test()
        
    except:
        import traceback
        traceback.print_exc()
        
if __name__=="__main__":main()