lightning과 비슷한 형식을 유지해 만들어 볼 것 - [ ] run - [ ] before training(=setup) - [ ] before epoch - [ ] before batch - [ ] after batch - [ ] after epoch - [ ] after training - [ ] test
lightning과 비슷한 형식을 유지해 만들어 볼 것