diff --git a/models.py b/models.py index 31350979..cfa4fff9 100755 --- a/models.py +++ b/models.py @@ -441,7 +441,6 @@ def attempt_download(weights): print(msg) os.system('rm ' + weights) # remove partial downloads - assert os.path.exists(weights), msg # download missing weights from Google Drive if os.path.getsize(weights) < 5E6: # weights < 5MB (too small), download failed os.remove(weights) # delete corrupted weightsfile - print(msg) + assert os.path.exists(weights), msg # download missing weights from Google Drive