• Shahriar's avatar
    C++ Models (#728) · b5db97b4
    Shahriar authored
    * Added the existing code
    
    * Added squeezenet and fixed some stuff in the other models
    
    * Wrote DenseNet and a part of InceptionV3
    
    Going to clean and check all of the models and finish inception
    
    * Fixed some errors in the models
    
    Next step is writing inception and comparing with python code again.
    
    * Completed inception and changed models directory
    
    * Fixed and wrote some stuff
    
    * fixed maxpoool2d and avgpool2d and adaptiveavgpool2d
    
    * Fixed a few stuff
    
    Moved cmakelists to root and changed the namespace to vision and wrote weight initialization in inception
    
    * Added models namespace and changed cmakelists
    
    the project is now installable
    
    * Removed some comments
    
    * Changed style to pytorch style, added some comments and fixed some minor errors
    
    * Removed truncated normal init
    
    * Changed classes to structs and fixed a few errors
    
    * Replaced modelsimpl structs with functional wherever possible
    
    * Changed adaptive average pool from struct to function
    
    * Wrote a max_pool2d wrapper and added some comments
    
    * Replaced xavier init with kaiming init
    
    * Fixed an error in kaiming inits
    
    * Added model conversion and tests
    
    * Fixed a typo in alexnet and removed tests from cmake
    
    * Made an extension of tests and added module names to Densenet
    
    * Added python tests
    
    * Added MobileNet and GoogLeNet models
    
    * Added tests and conversions for new models and fixed a few errors
    
    * Updated Alexnet ad VGG
    
    * Updated Densenet, Squeezenet and Inception
    
    * Added ResNexts and their conversions
    
    * Added tests for ResNexts
    
    * Wrote tools nessesary to write ShuffleNet
    
    * Added ShuffleNetV2
    
    * Fixed some errors in ShuffleNetV2
    
    * Added conversions for shufflenetv2
    
    * Fixed the errors in test_models.cpp
    
    * Updated setup.py
    
    * Fixed flake8 error on test_cpp_models.py
    
    * Changed view to reshape in forward of ResNet
    
    * Updated ShuffleNetV2
    
    * Split extensions to tests and ops
    
    * Fixed test extension
    
    * Fixed image path in test_cpp_models.py
    
    * Fixed image path in test_cpp_models.py
    
    * Fixed a few things in test_cpp_models.py
    
    * Put the test models in evaluation mode
    
    * Fixed registering error in GoogLeNet
    
    * Updated setup.py
    
    * write test_cpp_models.py with unittest
    
    * Fixed a problem with pytest in test_cpp_models.py
    
    * Fixed a lint problem
    b5db97b4
squeezenet.cpp 3.68 KB