Increase GPU memory usage

sensity-ai / dot

The Deepfake Offensive Toolkit

BSD 3-Clause "New" or "Revised" License

4.06k stars 396 forks source link

Increase GPU memory usage #159

Open Ghassen-Chaabouni opened 4 months ago

Ghassen-Chaabouni commented 4 months ago

:sparkles: Feature Request

Description:

Increase the usage of the GPU memory in order to speed up dot. Machines with low GPU memory should also be supported by switching to CPU if the memory is not enough.

Possible improvements:

Move all of the operations in this script reverse2original.py to GPU

IoT-VN commented 4 months ago

how to Increase with window version, my rog strix 3070 ti