search

Zhangjinso / PISE

123 stars 28 forks source link

I couldn't find the code locations that to "comments the line 177 and 178, and uncomments line 162-176." mentioned in README. #41

Closed dyyzb closed 5 months ago

dyyzb commented 11 months ago

Hi Jinsong, I'm really interest in PISE and admire your work. At your convenience, I'd like to confirm with you about the code locations that to "comments the line 177 and 178, and uncomments line 162-176." mentioned in the Train sector of README because I couldn't find these two pieces of code and I really want to study them. Looking forward to your reply. Thanks a lot!

Zhangjinso commented 11 months ago

you can find that in the generator file

dyyzb commented 11 months ago

I am very grateful for your reply. Also, I would like to ask if the training deadline is when the number of iterations reaches the threshold, and what is the approximate duration of model training. I really hope to receive your reply。

Zhangjinso commented 11 months ago

You can try it using the same parameters in other methods, e.g., 200 epochs. I remember I trained the model about 70k-80k iters on a single 2080Ti card.

dyyzb commented 11 months ago

Thank you very much for your reply. I also trained on 2080, with a set iteration count of 50k.