# I got a problem with the grade of assignment. Can't reach 100

**URL:** <https://community.deeplearning.ai/t/i-got-a-problem-with-the-grade-of-assignment-cant-reach-100/98793>\
**Category:** Introduction to TF for Artificial Intelligence ...\
**Tags:** week-module-2\
**Created:** [February 14, 2022, 9:14am UTC](https://community.deeplearning.ai/t/i-got-a-problem-with-the-grade-of-assignment-cant-reach-100/98793 "2022-02-14T09:14:12Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![Cactus](https://avatars.discourse-cdn.com/v4/letter/c/f475e1/32.png) [@Cactus](https://community.deeplearning.ai/u/Cactus)\
**Post date:** [February 14, 2022, 9:14am UTC](https://community.deeplearning.ai/t/i-got-a-problem-with-the-grade-of-assignment-cant-reach-100/98793/1 "2022-02-14T09:14:12Z")

</div>

![image](https://global.discourse-cdn.com/dlai/original/2X/0/08f56e7b219fdfd885933da09a9c3866a7d60706.png)  
I got this error.  
And here is my code.  
 ![image](https://global.discourse-cdn.com/dlai/original/2X/5/5f2612d89dd032f7be8c44022a9ffee0113bfe19.png)

---

<div class="post-metadata">

**Author:** ![a-zarta](https://sea1.discourse-cdn.com/dlai/user_avatar/community.deeplearning.ai/a-zarta/32/162492_2.png) [@a-zarta](https://community.deeplearning.ai/u/a-zarta)\
**Post date:** [February 14, 2022, 8:07pm UTC](https://community.deeplearning.ai/t/i-got-a-problem-with-the-grade-of-assignment-cant-reach-100/98793/2 "2022-02-14T20:07:01Z")

</div>

Hi @Cactus, try explicitly setting the input shape of the Flatten layer like `Flatten(input_shape=(28, 28))`. Right now your layer is inferring the first dimension from the default `batch_size` that the `fit` method has but it should be agnostic from the batch size used.