search

pytorch / android-demo-app

PyTorch android examples of usage in applications
1.46k stars 604 forks source link

Train model of video classification. #218

Open NeighborhoodCoding opened 2 years ago

NeighborhoodCoding commented 2 years ago

your code is great (video classification)

it works well in android, but in python code, the training part is missing.

I found

https://pytorchvideo.org/docs/tutorial_classification

to train it, but I have a question.

  1. after train,py, the ckpt model is made, can I use ckpt model to convert pt.mobile?
  2. train,py's some data preprocessing of data size seem may different, the input blob size may be not 3x160x160. cf. https://github.com/facebookresearch/pytorchvideo/issues/47 Could you provide full train code so I can made my model?

thx.