Hi,
I have successfully used it to train on single condition (with over 1k triplets) but while I have been trying it for multiple conditions then the training process (program) get stuck just before iterating over the test loader at line 237.
I'm just using toy dataset with around 8 triplets set files each containing 10 data points. I have made sure that images are present at their respective paths.
I'm using a Tesla V100-PCIE with 16GB GPU RAM. Pls let me know if any more info is required.
https://github.com/andreasveit/conditional-similarity-networks/blob/6021dfe5a7f3c0775a00d125e8de85982b651f43/main.py#L237
Hi, I have successfully used it to train on single condition (with over 1k triplets) but while I have been trying it for multiple conditions then the training process (program) get stuck just before iterating over the test loader at line 237.
I'm just using toy dataset with around 8 triplets set files each containing 10 data points. I have made sure that images are present at their respective paths.
I'm using a Tesla V100-PCIE with 16GB GPU RAM. Pls let me know if any more info is required.