• oahzxl's avatar
    Update environment.yml (#79) · ec9352d1
    oahzxl authored
    * Update environment.yml
    
    torch==1.11.1+cu113 is not supported in pip channel now
    
    * update other version
    
    * add torch version check
    
    if torch version > 1.11.1, turn on tf32
    ec9352d1
inference.py 19.4 KB