Closed jasam-sheja closed 3 years ago
Your work is fascinating.
Could you share some information on the environment you used, like OS (& version), python version, and libraries' versions (especially PyTorch)?
Also, when it comes to inference, How much GPU memory is required for a single image (let's say 512x512, same as your experiments)?
Thanks
the library version can see in the ./requirements.txt,and I set batchsize=4 on my 1080Ti for training.
Your work is fascinating.
Could you share some information on the environment you used, like OS (& version), python version, and libraries' versions (especially PyTorch)?
Also, when it comes to inference, How much GPU memory is required for a single image (let's say 512x512, same as your experiments)?
Thanks