rapidrabbit76 / Classification-For-Everyone

Classification with pytorch lightning
MIT License
3 stars 3 forks source link

MobileNetV3 PR #6

Closed zkdlfrlwl2 closed 2 years ago

zkdlfrlwl2 commented 2 years ago

MobileNetV3 PR & MobileNetV2 hyper-parameter tuning

zkdlfrlwl2 commented 2 years ago

list -> dict으로 바꿧는데 확인 좀 ㅎ

zkdlfrlwl2 commented 2 years ago

오늘 출근하기 전에 잠깐 확인해보니까 model save할 때, torch script에서 뭔가 Error 뱉던데 퇴근하고 확인 좀 해봐야할 듯

rapidrabbit76 commented 2 years ago

오늘 출근하기 전에 잠깐 확인해보니까 model save할 때, torch script에서 뭔가 Error 뱉던데 퇴근하고 확인 좀 해봐야할 듯

save 쪽은 리펙토링 할때 한번에 하자 지금은 빠르게 구현