cortex
cortex copied to clipboard
A machine learning library for PyTorch
Mostly adding .item() to kill warning and an old if that didn't work.
Using `.item()` to store results in `routine` call forces GPU to synchronize in order to have access at a lazy-evaluated Python number. This is suboptimal as kernel scheduling (CPU load)...
So right now, there is an init_fn being passed to the DataLoader to avoid a terminal flood when you do a keyboard interrupt. Normally, pytorch doesn't handle this well, but...
I have come to realize by the implementation in `built_ins/gan.py` that two separate calls are done in order to update the discriminator. The first optimizer update utilizes the gan loss...
Hello there! I would like to make a PR which would include the following things: - [x] Add an cmd line argument for a seed. - [x] Create a setter...
Either I do not understand how to use option `model_optimizer_options` or it will not work with current implementation. This PR allows to write `--o.model_optimizer_options "encoder={'lr': 0.01}"`.
Model build needs to be added to testing. Starting with classifier.py, with some variable arguments, the classifier needs to be tested for having the right parameter sizes as well as...