mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-04-10 12:33:44 +00:00
Initial commit
This commit is contained in:
9
ML/TensorFlow/CNN_architectures/LeNet5/test.py
Normal file
9
ML/TensorFlow/CNN_architectures/LeNet5/test.py
Normal file
@@ -0,0 +1,9 @@
|
||||
# disable tensorflow debugging messages
|
||||
import os
|
||||
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
||||
|
||||
from lenet5 import LeNet5
|
||||
|
||||
if __name__ == "__main__":
|
||||
model = LeNet5(input_shape = (32, 32, 1), classes = 10)
|
||||
model.summary()
|
||||
Reference in New Issue
Block a user