benchmark

e26be486 - Fix bug and disable tests that fail CI

Fix bug and disable tests that fail CI Disable CUDA on densenet121 due to CUDA OOM. Add NotImplemented to Init for densenet121 cuda. Disable CPU train and CPU+jit eval due to OOM