search

yhygao / CBIM-Medical-Image-Segmentation

A PyTorch framework for medical image segmentation
Apache License 2.0
260 stars 46 forks source link

3d UTNetV2模型 #7

Closed JiangZongKang closed 2 years ago

JiangZongKang commented 2 years ago

请问当训练大小为(128, 128, 128)时，使用3d UTNetV2模型训练，batch_size默认是多少啊，需要显存多少呢？

yhygao commented 2 years ago

In our experiment, we use batch_size 2. The memory consumption is about 39 GB. If you encounter OOM error, you may use the window_size (96, 96, 96).

JiangZongKang commented 2 years ago

In our experiment, we use batch_size 2. The memory consumption is about 39 GB. If you encounter OOM error, you may use the window_size (96, 96, 96).

thank you for your reply.