[colab] SystemExit :2 오류

    parser = argparse.ArgumentParser()

    parser.add_argument('--image_path', type=str, default="test_dirty_mnist/")

    parser.add_argument('--label_path', type=str, default="sample_submission.csv")

    parser.add_argument('--weight_path', type=str, default='/content/save/0927025311/model_best.pth',)

    parser.add_argument('--out_path', type=str, default='/content/drive/MyDrive/Dacon_hackathon')

    parser.add_argument('--model', type=str, default='efficientnet-b8')    

    parser.add_argument('--batch_size', type=int, default=4)

    parser.add_argument('--device', type=str, default=device)

    args = parser.parse_args()

 

 

위와 같은 내용(parser인자)은 주피터에서 사용한다.

 

따라서 코랩에서는 

 

args = easydict.EasyDict({

    "image_path": "/content/test_dirty_mnist",

    "label_path": "/content/sample_submission.csv",

    #"weight_path": os.path.join(SAVE_DIR, 'model_best.pth.tar'),

    "weight_path": '/content/save/0927025311/model_best.pth',

    'out_path': '/content/drive/MyDrive/Dacon_hackathon',

    "model": "efficientnet-b8",

    "batch_size": 8,

    "device": device,

})

 

다음과 같이 easydict를 사용해서 바꾸어주면 된다!