search

dptech-corp / Uni-Fold

An open-source platform for developing protein models beyond AlphaFold.
https://doi.org/10.1101/2022.08.04.502811
Apache License 2.0
376 stars 74 forks source link

Is total_step fixed? #120

Closed dingquanyu closed 1 year ago

dingquanyu commented 1 year ago

Hi,

Sorry again but could I double check on the setting of total_step? I saw it is hard-coded in the example script but I believe this number should be the same as the total number of entries in the train_sample_weight.json if batch_size is set to 1 right? I've changed it accordingly in my own case. I guess the user is also expected to change total_step case by case right ?

Thanks in advance.

ZiyaoLi commented 1 year ago

No. The dataset is randomly sampled instead of shuffled, so the total step is independent with dataset size.

dingquanyu commented 1 year ago

I see. Thanks for your explanation.