Skip to content
Snippets Groups Projects
Select Git revision
  • a9103600ea598d035f225859a708951a59a727db
  • master default protected
  • updated_emam2middleware
  • patch-1
  • corrected_get_torcs_image
  • develop
  • multi_variant_model
7 results

CNNTrainer_Dpnet.py

Blame
  • 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}
    
    
    
        )