Commit Graph

8 Commits

Author SHA1 Message Date
George Joseph
dc92dc7d8b Update notebook to fix issues with environment inheritance.
Two issues:

* The notebook cell that actually runs model_train_eval was running it in a
  subprocess so while it inherited environment variables from the running
  python kernel, it couldn't inherit the tensorflow environment from it.
  This resulted in the `set_memory_growth(g, True)` and
  `mixed_precision.set_global_policy("mixed_float16")` calls in the previous
  cell to be lost.

* TFlite doesn't support "mixed_float16" anyway and causes the model export to
  fail spectacularly so it's kind of a good thing it wasn't being applied.

So..

* The tensorflow environment variable and memory_growth setting code was moved
  from the notebook cell that also wrote the config yaml to the next cell
  which does the train and test.  This leaves the "config" cell to just write
  the yaml.  This is really just a cosmetic change to group functionality
  better.

* The code that tried to set "mixed_float16" has been removed but since setting
  memory_growth to true is a good thing, the model_train_eval is now run using
  runpy instead in a subprocess.  This way it's run in the same python kernel
  instance and tensorflow environment as the rest of the notebook and inherits
  the memory_growth setting.

Resolves: #14
2025-12-20 10:22:33 -07:00
Tater Totterson
efcf8a4add update 2025-11-02 09:53:03 -06:00
Tater Totterson
9ad63fef34 fixes 2025-11-02 05:08:13 -06:00
Tater Totterson
31677fa74b fixes
thanks @ mitrokun
2025-11-02 04:44:24 -06:00
Tater Totterson
763fe7409a cleanup 2025-09-27 15:26:43 -05:00
Tater Totterson
cef3daa1a7 cleanup 2025-09-27 14:55:23 -05:00
Tater Totterson
e9d6b8b87e cleanup 2025-09-27 11:10:43 -05:00
Tater Totterson
afa5adb794 Add files via upload 2025-09-26 19:36:45 -05:00