-
We have a use case for the "on_predict_epoch_end" callback (https://lightning.ai/docs/pytorch/stable/extensions/callbacks.html#on-predict-epoch-end) in NMF. We would like to save the learned gene fac…
-
I'm still new to this codebase and may not have a perfect understanding or description of how the engine handles imported functions or epoch canceling. Any feedback is appreciated.
#### Improvement…
-
I was doing my learning which needs many single_train_steps on my gpu(NVIDIA GeForce RTX 2060 Mobile) when I noticed the irregular RAM(not video RAM) usage. I tested by modifying the epochs of https:/…
-
想问一下作者大概训练了多少个epoch,花了多长时间?
我为了在2080TI(22G)上训练把train_batch_size设置成了1,一个epoch貌似要16个小时....
还有就是想问一下max_train_steps是总共只有30000step还是一个epoch有30000step
-
after discussion, look like epochs are not really needed, we can directly use batches. so going from "round -> epoch -> batch" to have "round -> batch". that would give more direct control on
* [ ]…
-
I am planning to create a model to detect eyes, nose, mouth, and ears in 3D fMRI data for the purpose of defacing. I intend to train this model using MedYOLO. My dataset consists of 250 nii.gz files p…
-
After annotating a couple of samples, when I switch to the refine strategy, I get an error where the shapes of tensors do not match:
```
(venv) bengt@bengt-desktop:~/Downloads/DFKI/github.com/yapa…
Bengt updated
17 minutes ago
-
We need to estimate when the model starts overfitting during pretraining. We should have a table like this:
Epoch | Training Total | Validation Total | Training MLM | Validation MLM | Training C…
-
I wonder how I can use LiteEFG to implement an algorithm where the epoch updates are not homogeneous. For example, I want to implement an algorithm that has a pivot point updated every 100 epochs.
…
-
I'm trying to run the benchmark but it crashes on the `dcase2016_task2` task. After training for what seems like 229 epochs, at the prediction stage, I get a `KeyError` trying to access the postproces…