Hello! Thanks for the work done, the results look great. I was particularly impressed by your image inversion, but I am not quite sure how it works. Do you plan to publish the relevant code?

I used to successfully train the StyleGAN2-ADA and StyleGAN3 on my device. However the distributed training for SOAT failed due to out of the cuda memory. I modify the code a little bit which don't involving any training codes, then I use the Slurm to submit my training job to the server and check the model has been successfully distributed to different GPUs. Before the first epoch completes, the job aborts.
The information below is my training environment:
    CPU: Intel Xeon 6348
    GPU: NVIDIA A100 40G PCIe*8
    Script:  python -m torch.distributed.launch --nproc_per_node=8 train.py --dataset=[My Dataset(Grayscale in 1024x1024, and I convert it into RGB when loading dataset)] --batch=X --size=1024 --iter=40000
BTW, I set the batch size as 64, 32, 16. All of them abort. When I using a single GPU to train the SOAT with batch size 8, it succeeds.
Looking for your reply and see if there's any possible solution.

I'd like to annotate and transfer multiple bboxes from the target image to the source image. I noticed in running infinity.ipynb that while I can use Colab to annotate multiple bboxes on the target, only the first of them gets transferred from the source.

This behavior seems consistent with the blend_bbox code in model.py where only the first bbox (coord[0]) is being considered and the rest don't seem to be processed.

Two questions - a) am I missing something here and b) if the above is accurate, is it useful if I add an outer loop to enable blend_bbox to iterate thru the bbox_list

When I start running the colab file, it stop responding in

freeman@freeman-T430s:~/SOAT/models/research$ python -m pip install -q .
  DEPRECATION: A future pip version will change local packages to be built in-place without first copying to a temporary directory. We recommend you use --use-feature=in-tree-build to test your packages with this new behavior before it becomes the default.
   pip 21.3 will remove support for this functionality. You can find discussion regarding this at https://github.com/pypa/pip/issues/7555.

^CERROR: Operation cancelled by user

It occurs on my local laptop too. What should I do ?