search

tzayuan / TransMatch_TMI

This repository implements the TransMatch model for unsupervised deformable image registration, as published in the IEEE TMI journal.
40 stars 4 forks source link

out of memory when training in OASIS dataset #8

Open Runshi-Zhang opened 6 months ago

Runshi-Zhang commented 6 months ago

When TransMatch was run in OASIS dataset(using your LPBA40_config and the input size (160, 192, 224)), the GPU memory was over 24GB in NVIDIA GeForceRTX 3090Ti GPUs (24 GB of VRAM).

ZhaiJiaKai commented 2 months ago

在 OASIS 数据集中运行 TransMatch 时（使用您的LPBA40_config和输入大小 （160， 192， 224）），NVIDIA GeForceRTX 3090Ti GPU 中的 GPU 内存超过 24GB（24 GB 的 VRAM）。

I also encounter the same problem, have you solved it?