jwyang/JULE.torch

It is difficult to train in Large dataset.

Closed this issue · 3 comments

I use 80000 samples to train the jointed net. But when I finished the first CNN update, it is difficult to run the next step. This code seemly have a large amount of computation in computing the 'Affinity',
How can I solve this problem?

Thanks for your advice.
I attempt to compute partial affinity by spliting NNs to batches like you did in batch knn. But I failed. Because NMI for MNIST-test is much lower than before.
So how can I correctly get the partial affinity.

Hi, I think one way to solve this is using some fast knn algorithm to build connections for close samples, and then compute the affinity for these close samples.