Solved by chetan_patil in post #2 
              
              
                Since you are hitting an error in FC layer, the problem is in the input size, could you change the input sizes to (1,3,224,224) or (1,3,229,229) ?