Submitted by aleguida t3_yde1q8 in MachineLearning
aleguida OP t1_itvgiu8 wrote
Reply to comment by seba07 in [D] Tensorflow learning differently than Pytorch by aleguida
Thanks for the feedback. I tried retraining everything from scratch without downloading any pretrained weights. here is the colab links update.
- TF: https://colab.research.google.com/drive/1O6qzopiFzK5tDmLQAzLKmNoNaEMDc4Ze?usp=sharing
- PYTORCH: https://colab.research.google.com/drive/1g-1CEpzmWJi9xOiZHzvDSv_-eDlXZO9u?usp=sharing
​
While Pytorch is learning something, Tf is not learning anything. This is actually quite confusing as I used tf.Keras to minimize any possible error on my part. I will try to build the same network from scratch in both frameworks next
[deleted] t1_itzwb38 wrote
[deleted]
Viewing a single comment thread. View all comments