Commit Graph

5 Commits

Author SHA1 Message Date
Stefan Saraev
b5a5f72eda load speedup: refactored image loading
Images are now loaded on the target device as uint8s.
Then they are converted to the target data type (eg. fp32 or fp16).
This speeds up the loading time.

Also, users can opt to store the image as uint8 or as target data type.
This will further reduce memory usage.
2024-05-15 22:32:58 +03:00
Stefan Saraev
18eb6d6a0c added support for setting floating point range
Users may want to reduce their memory consumption by using fp16.
However, in my tests, such attempts will result in lower quality renders.
Some data type conversions did not have any impact, so I removed them completely.
2024-05-15 22:29:23 +03:00
Pythonix Huang
989320fdf2
Provide --data_device option to put data on CPU to save VRAM for training (#14)
* Provide --data_on_cpu option to save VRAM for training

when there are many training images such as in large scene, most of the VRAM are used to store training data, use --data_on_cpu  can help reduce VRAM and make it possible to train on GPU with less VRAM

* Fix data_on_cpu  effect on default mask

* --data_on_cpu to --data_device

* update readme

* format warning infos
2023-07-12 11:30:45 -07:00
Bernhard Kerbl
26942586b1 Added licenses, bumped submodule versions 2023-07-05 12:22:16 +02:00
bkerbl
15d64e6781 Initial commit 2023-07-04 10:00:48 +02:00