boahk/darl

Official repository of "Diffusion Adversarial Representation Learning for Self-supervised Vessel Segmentation", ICLR 2023.

darl's Issues

How can I run the code on multi-GPUs

Thank you for your wonderful work. When I run the code, I try to allocate the model and data into 2 GPUs but it only runs on the first GPU.
I think the possible reason is that set_device function in Base_model moves data to one GPU.
Could you please tell me how to change it to run on multi-GPUs?
Thanks a lot!!

HOW TO SOLVE IndexError: list index out of range

Thank you for open sourcing your code, there was a problem in the process of replicating your code. when I run the command:
python3 main.py -p train -c config/train.json.
IndexError: list index out of range
I have already used two GPUs, I hope I can get your help

code for C-DARL

Dear boahK,

first of all, thanks for the nicely documented repo and your work on self-supervised vessel segmentation!
Recently, I came across your follow-up work C-DARL and was wondering whether you are planning on releasing the code for this extension? This would be highly appreciated.

Best and thanks in advance,
Bastian