siddharthsinh Yadav comments

Repositories
Issues
Comments

Results 2 comments of


                                            siddharthsinh Yadav

I finetuned the model for 3 output classes but now at the time of prediction i got the error of input Shape does not match

` import argparse import numpy as np import tensorflow as tf import i3d import cv2 def main(IMAGE_SIZE=224,TOP_RESULT_CLASSES=3,SAMPLE_VIDEO_FRAMES = 250,\ SAMPLE_PATH_RGB = '../input/v_CricketShot_g04_c01_rgb.npy',\ SAMPLE_PATH_VIDEO= '../../finetune_i3d/input/test_vid/Abuse/Abuse001_x264.mp4',\ SAMPLE_PATH_FLOW = '../input/v_CricketShot_g04_c01_flow.npy',\ EVAL_TYPE = 'rgb',imagenet_pretrained...

What minimum amount of GPU is needed to finetune pretrained ? 8gb GPU is enough? or i need 12 gb GPU?

Thanks for the help @joaoluiscarreira