• LXYTSOS's avatar
    utils.py in references can't work with pytorch-cpu (#1023) · 250bac89
    LXYTSOS authored
    * can't work with pytorch-cpu fixed
    
    utils.py can't work with pytorch-cpu because of this line of code `memory=torch.cuda.max_memory_allocated()`
    
    * can't work with pytorch-cpu fixed
    
    utils.py can't work with pytorch-cpu because of this line of code 'memory=torch.cuda.max_memory_allocated()'
    250bac89
utils.py 8.84 KB