search

GXYM / TextBPN-Plus-Plus

Arbitrary Shape Text Detection via Boundary Transformer；The paper at: https://arxiv.org/abs/2205.05320, which has been accepted by IEEE Transactions on Multimedia (T-MM 2023).
172 stars 37 forks source link

What version of Pytorch do you use? #2

Open YTMartian opened 2 years ago

YTMartian commented 2 years ago

Thanks for your job. Could you please tell me which version of Pytorch do you use? Because some code not work in pytorch 1.11.0,thanks.

GXYM commented 2 years ago

We test the environment “ Ubuntu20.04+Python3.8 PyTorch >= 1.10.0-cu113;”

In the this environment, you may be need to use a new parameter setting in the “data.DataLoader”，such as train_loader = data.DataLoader(trainset, batch_size=cfg.batch_size, shuffle=True, num_workers=cfg.num_workers, pin_memory=True, generator=torch.Generator(device=cfg.device))

I hope this will help you！

YTMartian commented 2 years ago

We test the environment “ Ubuntu20.04+Python3.8 PyTorch >= 1.10.0-cu113;” In the this environment, you may be need to use a new parameter setting in the “data.DataLoader”，such as train_loader = data.DataLoader(trainset, batch_size=cfg.batch_size, shuffle=True, num_workers=cfg.num_workers, pin_memory=True, generator=torch.Generator(device=cfg.device)) I hope this will help you！

Thanks for your kindly help, it works.