rA9
rA9 copied to clipboard
Performance Enhancement
img2col.py in nn/_functions/ dataloader.py in utils/data/
Because some parts of img2col use numpy not jax.numpy, and dataloader is based on python so it is quite slow
#75 the onp.add.at is changed into index add so it is enhanced. signed a die
Dataloader problem remained
@jepetolee will solve this issue
Remove the input_elements class so it will be more faster than before
Encoder is slow somehow
input layer has returned, that can be used as encoder