Issues :: PyTorch .h 학습 모델을 ONNX로 변환 안되는 문제 해결하기
Issues :: PyTorch .h 학습 모델을 ONNX로 변환 안되는 문제 해결하기 해당 문제로 2-3일 정도 사용해서 문제를 해결했다일단 내가 하고싶었던 코드는 개랑 고양이를 구별하는 학습모델(.h5) 를 ONNX로 변환하려고 하는데# test-modelData.pyimport tensorflow as tffrom tensorflow.keras.models import Sequentialfrom tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropoutfrom tensorflow.keras.preprocessing.image import ImageDataGeneratorimport os# GPU 메모리 설정gpus = ..