flax
25ee4ce5 - Enhance CNN model with BatchNorm and Dropout layers for improved training stability and performance. Update training and evaluation steps to switch model modes appropriately. Clarify documentation on the use of .eval() for inference.

Commit

188 days ago

Enhance CNN model with BatchNorm and Dropout layers for improved training stability and performance. Update training and evaluation steps to switch model modes appropriately. Clarify documentation on the use of .eval() for inference.

References

#4773 - Add BatchNorm layers to CNN in MNIST tutorial for improved training stability

Author

sanepunk

Parents

96c565ff

flax 25ee4ce5 - Enhance CNN model with BatchNorm and Dropout layers for improved training stability and performance. Update training and evaluation steps to switch model modes appropriately. Clarify documentation on the use of .eval() for inference.

flax
25ee4ce5 - Enhance CNN model with BatchNorm and Dropout layers for improved training stability and performance. Update training and evaluation steps to switch model modes appropriately. Clarify documentation on the use of .eval() for inference.