파이토치에서 모델을 불러와 예측하는것에 대해 질문드립니다.

조회수 56회

범죄 감지 CCTV를 만들고있는데, 모델을 불러와 예측을 하는과정에서 막혀서 질문드립니다. 모델은 SlowFast를 사용하고있습니다.

model.load_state_dict((torch.load('weights_crime/c3d_19.h5')))

이렇게 모델을 불러오고

model.predict(Dataset)

이렇게 예측을 할려하면

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-83-fde4eee11ca9> in <module>
----> 1 model.predict(Dataset)

D:\anaconda\envs\mj\lib\site-packages\torch\nn\modules\module.py in __getattr__(self, name)
   1128             if name in modules:
   1129                 return modules[name]
-> 1130         raise AttributeError("'{}' object has no attribute '{}'".format(
   1131             type(self).__name__, name))
   1132 

AttributeError: 'SlowFast' object has no attribute 'predict'

이런 오류가 발생합니다.

답변을 하려면 로그인이 필요합니다.

Hashcode는 개발자들을 위한 무료 QnA 사이트입니다. 계정을 생성하셔야만 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)

ᕕ( ᐛ )ᕗ
로그인이 필요합니다

Hashcode는 개발자들을 위한 무료 QnA사이트 입니다. 계정을 생성하셔야만 글을 작성하실 수 있습니다.