BaseModel().cuda() out of memory

[PhoenixZ810]

When I run
def build_model(cls, cfg):
model = BaseModel(cfg, build_model(cfg)).cuda()
in train_net.py
the memory-usage will increase without stop. Finally cuda out of memory.
I have tried lots of times and I don't know where it comes from.

[HaoZhang534]

Hello, can you provide more information to help me reproduce your error, such as the command you use?