Nitish Shirish Keskar 8becb73293 removing torch.cuda.empty_cache() from TF function (#3267) ...

torch.cuda.empty_cache() was being called from a TF function (even when torch is unavailable)
not sure any replacement is needed if TF OOMs

2020-03-19 23:25:30 +01:00

28 KiB

Raw Blame History

View Raw