Self-contained checkpoint --resume (#8839)
* Single checkpoint resume * Update train.py * Add hyp * Add hyp * Add hyp * FIX * avoid resume on url data * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * avoid resume on url data * avoid resume on url data * Update Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
1 parent
4d8d84b0ea
commit
a75a1105a1
3 files changed
+27
-15
No files matched your search
+6
-4
@@ -16,12 +16,14 @@ import requests
|
||||
import torch
|
||||
|
||||
|
||||
def is_url(url):
|
||||
def is_url(url, check_online=True):
|
||||
# Check if online file exists
|
||||
try:
|
||||
r = urllib.request.urlopen(url) # response
|
||||
return r.getcode() == 200
|
||||
except urllib.request.HTTPError:
|
||||
url = str(url)
|
||||
result = urllib.parse.urlparse(url)
|
||||
assert all([result.scheme, result.netloc, result.path]) # check if is url
|
||||
return (urllib.request.urlopen(url).getcode() == 200) if check_online else True # check if exists online
|
||||
except (AssertionError, urllib.request.HTTPError):
|
||||
return False
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user