Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # initialize our initial learning rate, # of epochs to train for,
- # and batch size
- INIT_LR = 0.01
- EPOCHS = 75
- BS = 32
- # initialize the model and optimizer (you'll want to use
- # binary_crossentropy for 2-class classification)
- print("[INFO] training network...")
- opt = SGD(lr=INIT_LR, decay=INIT_LR / EPOCHS)
- model.compile(loss="categorical_crossentropy", optimizer=opt,
- metrics=["accuracy"])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement