MoViNet-pytorch icon indicating copy to clipboard operation
MoViNet-pytorch copied to clipboard

Modifying for binary classification

Open ekuluke opened this issue 3 years ago • 2 comments

Hi, trying to solve a binary classification problem with these movinets and was wondering how I would go about creating a MoVinet with some augmentation with the aim of reducing the TFLOPS required and decreasing the inference time

ekuluke avatar Dec 11 '21 06:12 ekuluke

Hey, have you solved your binary classification problem? I'm also working with MoViNets and binary classification atm and have a few questions.

kaethebu avatar Mar 30 '22 09:03 kaethebu

Hi, using (2+1)D convolution can reduce computation compared with 3D convolution. But I think the stream model should be well designed to decrease the inference time. Also, the stream buffer is returned by the model and sent back to the model in TF version, but that is stored inside the model in this PyTorch implementation. Maybe TF version is a good choice?!

patrick620 avatar Apr 15 '22 09:04 patrick620