Commit Graph

12 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
bkerbl
f11001b46c Random background flag added 2023-10-23 16:02:30 +02:00
bkerbl
d4fa4779d5 Debugging functionality 2023-07-23 12:48:17 +02:00
bkerbl
6891d458f9 Scaling LR fixed 2023-07-16 11:48:04 +02: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
Pythonix Huang
f4dfb3d3bf Fix spelling error (posititon to position as README said) 2023-07-11 18:16:07 +08:00
bkerbl
490fab14dd Auto rescaling, including warning and documentation 2023-07-08 18:20:08 +02:00
bkerbl
0327061ffd Absolute path happens later 2023-07-07 01:54:20 +02:00
bkerbl
4268e86b79 Use absolute paths, update readme 2023-07-06 16:59:35 +02: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