Commit Graph

8 Commits

Author SHA1 Message Date
bopan3
933aa7d210 fix bug for mask loading 2023-09-07 21:19:47 +08: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
bkerbl
5776ed5e4e updated readme 2023-07-09 17:39:19 +02:00
bkerbl
29053b7cb0 it's k, not m 2023-07-09 16:08:01 +02:00
bkerbl
424012cd0d Changes to README, rescale from 1.6mpix 2023-07-09 14:36:02 +02:00
bkerbl
490fab14dd Auto rescaling, including warning and documentation 2023-07-08 18:20:08 +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