search

cyclomon / UNSB

Official Repository of "Unpaired Image-to-Image Translation via Neural Schrödinger Bridge" (ICLR 2024)
MIT License
154 stars 5 forks source link

Some problems in reproduction #20

Open Jungle12138 opened 4 months ago

Jungle12138 commented 4 months ago

Thank you for your wonderful work! I recently ran into some questions replicating your model on my own dataset and would like your answers.

  1. Limited by the fact that my GPU only has 16g of video memory, the GPU memory usage in the test stage reached 20G+ (borrowed part of the system memory), while the GPU memory usage in the training stage was less than 10G(did not borrow the system memory), is this situation normal?
  2. Since NFE=1,2,3,4,5 pictures in total will be generated for each picture in the test stage, which is very time-consuming, may I ask where to modify the code to generate only pictures such as NFE=5? I am looking forward to your answer. Thank you