mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-20 13:50:41 +00:00
10 lines
233 B
Python
10 lines
233 B
Python
|
# 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()
|