현재 준 코드 바탕으로 실험(vit)
메모리 이슈 있을 수 있음
그 전에 할 것 이진분류- > 멀티로.. (때 되면 알려주심.)