pytorch 회전 onnx 주의사항 (번역)

2138 단어 pytorchonnx
  • 데이터 [index] = new데이터와 같은 장량은 현재 내보내기에서 지원되지 않습니다.이런 문제를 해결하는 방법의 하나는 산자 산점을 사용하여 원시 장량을 현저하게 갱신하는 것이다.바로 tensorflow의 정적 그림처럼 tensor의 값을 마음대로 바꿀 수 없으며 torch의scatter오류 해결 방법
    • # def forward(self, data, index, new_data):
#     data[index] = new_data          #  
#     return data

    올바른 방식
    class InPlaceIndexedAssignmentONNX(torch.nn.Module):
    def forward(self, data, index, new_data):
        new_data = new_data.unsqueeze(0)
        index = index.expand(1, new_data.size(1))
        data.scatter_(0, index, new_data)
        return data
  • LSTM과 같은 동적 변수의 클래스를 장착
    • Pyfolio를 사용해볼게요.
    Requests에 대한 urlencode 문제

    좋은 웹페이지 즐겨찾기

    개발자 우수 사이트 수집

    개발자가 알아야 할 필수 사이트 100선 추천 우리는 당신을 위해 100개의 자주 사용하는 개발자 학습 사이트를 정리했습니다
    best100-homepage