[Feat] Add model-load reslves #20

[Lim-JH-Laskaris]

개요

• 학습끝난 pth 불러와서 이어서 학습할 수 있는 기능 추가

작업사항

• train.py에 model load 코드 추가
• train.py에 aurg로 --load 추가

    #to load parameters
    parser.add_argument('--load', type=str, default = "F")

    def do_training(~~~, **load**):  

    ################################모델 불러오기 코드 #############################################
    if load != "F":
        if   load =="T": #####정균님 0.67 코드 불러오기 ######
            print('load parameter from /baseline_200_(ICDAR17_all).pth')
            model.load_state_dict(torch.load(model_dir+"/baseline_200_(ICDAR17_all).pth"))

        elif load =="L": ##### 학습하던 코드 불러오기(exp_name 그대로인 경우)#####
            print('load parameter from latest.pth')
            model.load_state_dict(torch.load(model_dir+f'/{exp_name}_latest.pth'))
        else :           ##### pth 지정하기 (단, 학습하던 모델 디렉터리에 존재해야함) #####
            print(f'load parameter {load}')
            model.load_state_dict(torch.load(model_dir+'/'+load)) 
    ############################################################################################

완료된 Task

• train.py에 model load 코드 추가
• train.py에 aurg로 --load 추가

resolved #20