tinygrad/extra/optimization
George Hotz 4124cf1df5
cleanup tensor cores, expose exclude local upcast (#2064)
* expose exclude_local_upcast

* convert apply tensor cores to ops

* update comment

* put LOCAL back to what it was, BEAM is better than way
2023-10-14 09:21:03 -07:00
..
generate_dataset.sh op logger + replay (#2021) 2023-10-08 15:10:18 -07:00
get_action_space.py cleanup tensor cores, expose exclude local upcast (#2064) 2023-10-14 09:21:03 -07:00
helpers.py train value net, improve API, add BCE (#2047) 2023-10-12 07:56:38 -07:00
pretrain_policynet.py train value net, improve API, add BCE (#2047) 2023-10-12 07:56:38 -07:00
pretrain_valuenet.py train value net, improve API, add BCE (#2047) 2023-10-12 07:56:38 -07:00
rl.py train value net, improve API, add BCE (#2047) 2023-10-12 07:56:38 -07:00
test_net.py train value net, improve API, add BCE (#2047) 2023-10-12 07:56:38 -07:00
test_time_linearizer.py with unroll, the action space goes from 161 -> 127 (#2060) 2023-10-12 20:52:23 -07:00