Closed Chadkowski closed 1 year ago
even when running base alone and small alone still overflow. I changed channels to 1 instead of 4 in Swin3D but to no avail
Hi, you cannot load both models at once for training. Can you check whether any pids are listed under your GPU? Try this command : sudo fuser -v /dev/nvidia0
What is the dataset you are trying to train here? If its brats, then input channel size is 4.
Managed to sort it by changing patch size to 64/128/128 for my set but now I have the following issue..
2022-07-22 15:28:50.920116:
epoch: 0
Traceback (most recent call last):
File "/home/dawidr/.local/bin/vtunet_train", line 33, in
Hi, the current version of VTUNet can only handle the input size of Cx128x128x128. Therefore, the network architecture should be modified accordingly to process the input with the size of Cx64x128x128.
Hi! I'm still having issues with overflowing my VRAM (RTX 3090 24GB) whenever attempting to train. Even after I've redued my batch size to 1. Any ideas on what I can do?