Tests for MLP Mixer #63
Conversation
All modification requested All modification requested Modified type of weights Added init.py Added MLPMixer Model
|
Hi @juliagsy, can you please review this PR. I have added the test for the MLP Mixer Model! |
|
Hey! Just to let you know I'm reviewing the PR already but am looking into a failure, if it's related to ivy or the pr itself, will get back to you soon! thanks! |
| # Value test | ||
| if load_weights: | ||
| np_out = ivy.to_numpy(logits) | ||
| true_indices = np.array([1, 6, 9, 7, 4]) |
There was a problem hiding this comment.
Hey! I've briefly helped out with some permutations updates. Could you please double check the indices and logits because the values are floating pretty much atm, thanks!
|
Hi @juliagsy, I have made the changes. Could you please review it 😄 |
There was a problem hiding this comment.
Hey! Thanks for the changes! I'm not sure if it's happening on your side, but I'm on latest docker and latest models, there seem to be an unidentical structure in weights (missing dropout attr), would you mind to upload the updated weights to hf again and test it out? (which means the tests are failing, models couldn't be loaded as well) thanks a lot!
|
Hi @juliagsy, I have updated the weights. Can you please review it |
|
Thanks for the great work! |
Adding tests for MLP Mixer