Source Code Docs

• CNN Module
  • CNN
    • CNN.create_and_train_cnn()
    • CNN.create_criterion()
    • CNN.create_model()
    • CNN.create_optimizer()
    • CNN.evaluate_model()
    • CNN.train_epoch()
    • CNN.train_model()
• Main Module
  • define_transforms()
  • main()
  • read_images()
• Config Module
  • Config
    • Config.cpuUsed
    • Config.trainPath
    • Config.testPath
    • Config.validationPath
    • Config.modelsPath
    • Config.transformsHeight
    • Config.transformsWidth
    • Config.replications
    • Config.batchSize
    • Config.modelNames
    • Config.epochs
    • Config.learningRates
    • Config.weightDecays
    • Config.tree
    • Config.readConfigFile()