Zhicheng Chen

My game also crash because of OOM when switching between different songs. However, in a small number of cases, the game crashes with sufficient memory (> 1000MB). Don't know why,...

"Thanks for your constructive suggestions, we will consider add some logic to resolve this problem, thank you!" ---- @jacklightChen

I made a docker image with preconfigured environment, it works fine on my computer, you can have a try. https://hub.docker.com/r/4f5da2/person_search

According to the [documentation of DistributedSampler](https://pytorch.org/docs/stable/data.html?highlight=distributedsampler#torch.utils.data.distributed.DistributedSampler), I made following changes to `trainer.py`: ``` @@ -75,6 +75,9 @@ def get_trainer(args, model, model_without_ddp, train_loader, optimizer, lr_sche @trainer.on(Events.EPOCH_STARTED) def _init_epoch(engine): + if args.distributed:...