MSRF-Net icon indicating copy to clipboard operation
MSRF-Net copied to clipboard
verified publisher icon NoviceMAn-prog

fix: Left last batch out of training

Open augustoicaro opened this issue 3 years ago • 0 comments

This PR fixes some images left in batch training when the following condition is never met due to float to int conversion.

https://github.com/NoviceMAn-prog/MSRF-Net/blob/60e23734d94d340b9d89b64e7f9e2636b6f96c0f/train.py#L103

Also, used tqdm dependence that was never used.

augustoicaro avatar Feb 15 '22 03:02 augustoicaro