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/AlexNet/test.py
Normal file
9
ML/TensorFlow/CNN_architectures/AlexNet/test.py
Normal file
@@ -0,0 +1,9 @@
|
||||
# disable tensorflow debugging messages
|
||||
import os
|
||||
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
||||
|
||||
from alexnet import AlexNet
|
||||
|
||||
if __name__ == "__main__":
|
||||
model = AlexNet(input_shape = (224, 224, 3), classes = 1000)
|
||||
model.summary()
|
||||
Reference in New Issue
Block a user