Skip to main content

All Questions

Filter by
Sorted by
Tagged with
0 votes
1 answer
210 views

Training the network with some batch size - code

There is my "training" code below, I wrote it based on one youtube tutorial. I don't understand actually one part: batch_X = train_X[i:i+BATCH_SIZE], batch_y = train_y[i:i+BATCH_SIZE]. How ...
0 votes
1 answer
736 views

UNet Model accuracy is stuck at exact 0.5 (neither more or less) (No class imbalance, tried tuning learning rate)

This is using PyTorch I have been trying to implement UNet model on my images, however, my model accuracy is always exact 0.5. Loss does decrease. I have also checked for class imbalance. I have ...