nsml.load

nsml.load(checkpoint, load_fn=None, session=None)

저장된 모델을 불러옵니다. nsml.bind() 함수에서는 load 함수를 정의만 합니다.

매개변수

• checkpoint (str) – session 의 호출할 checkpoint을 정의합니다.

• load_fn (fn) –

  모델을 load 하는 방법이 정의된 함수를 인수로 받습니다. load 함수가 호출되는 우선순위는

  1. 이 함수로 전달된 load_fn

  2. nsml.bind() 에 binding 된 load 함수

  3. nsml 에서 정의한 default load 함수입니다.

• session (str) – 해당 session name 에서 저장된 model 을 불러옵니다.

Example

if __name__ == '__main__':
    ...
    if config.mode == 'train':
        ...
        # write down model number on checkpoint and session name that want to load.
        nsml.load(checkpoint)='model number', session='session name')