Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CNN in pytorch #90

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Conversation

GouravWadhwa
Copy link

Hi @nidhaloff,
I have added the feature for CNN in pytorch where the user enters random model. I also created a igel_cnn.yaml file for example.
I have also tested it on my local machine and is able to clear all the tests. I have also tested the evaluate function locally.

@GouravWadhwa
Copy link
Author

This PR solves the issue #76.

@GouravWadhwa
Copy link
Author

Hi @nidhaloff, I added more functionality in the pytorch code and now following activation layers are supported [relu, sigmoid, leaky relu, prelu, tanh, elu, hardshrink, hard_sigmoid, hard_tanh, hard_swish, log_sigmoid, multi_head_attention, relu6, selu, celu, gelu, silu, mish, softplus, softshrink, softsign, tanshrink, threshold, softmax2d, softmax, softmin, adaptive_softmax,log_softmax]
Further now I also added the following normalisation for users
[batch_norm1d, batch_norm, batch_norm3d, lazy_batch_norm1d, lazy_batch_norm, lazy_batch_norm3d, sync_batch_norm, instance_norm1d, instance_norm, instance_norm3d]
Further I also extended the support for types of pooling layer
[max_pool, avg_pool, adaptive_max_pool, adaptive_avg_pool]
Next, I also integrated the evaluate function in the Igel and also tested the functionality locally.

@GouravWadhwa GouravWadhwa mentioned this pull request Aug 27, 2021
@nidhaloff
Copy link
Owner

@GouravWadhwa Hi, sorry I was very busy and somehow I didn't notice your PR. I will check it out asap. Can you add an example on how to use your pytorch cnn in the examples folder? and/or maybe describe a simple example in README. Thanks for your PR

@GouravWadhwa
Copy link
Author

Hi @nidhaloff,
I have added examples of using CNN. You just need to download the mnist_png dataset (From the site you gave the description earlier).
It will store the results in model_results from where you can further evaluate it too.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants