Select Git revision
Project.sln
Code owners
Assign users and groups as approvers for specific file changes. Learn more.
CNNTrainer_Dpnet.py 658 B
import logging
import mxnet as mx
import CNNCreator_dpnet
if __name__ == "__main__":
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger()
handler = logging.FileHandler("train.log","w", encoding=None, delay="true")
logger.addHandler(handler)
dpnet = CNNCreator_dpnet.CNNCreator_dpnet()
dpnet.train(
batch_size = 64,
num_epoch = 100,
context = 'cpu',
normalize = True,
optimizer = 'sgd',
optimizer_params = {
'weight_decay': 5.0E-4,
'learning_rate_minimum': 0.01,
'learning_rate_decay': 0.9,
'step_size': 8000}
)