Commit
·
c9d4907
1
Parent(s):
fa98f1c
Upload Upstream: comit message: my best model
Browse files
{{cookiecutter.repo_name}}/cli.py
CHANGED
|
@@ -37,24 +37,23 @@ def validate():
|
|
| 37 |
for task in tasks:
|
| 38 |
downsample_rate = upstream.get_downsample_rates(task)
|
| 39 |
assert isinstance(downsample_rate, int)
|
| 40 |
-
|
| 41 |
-
f" has the downsample rate of {downsample_rate}.")
|
| 42 |
except:
|
| 43 |
print("Please check the Upstream Specification on https://superbbenchmark.org/challenge")
|
| 44 |
raise
|
| 45 |
|
| 46 |
typer.echo("All submission files validated!")
|
| 47 |
-
typer.echo("Now you can
|
| 48 |
|
| 49 |
|
| 50 |
@app.command()
|
| 51 |
-
def
|
| 52 |
subprocess.call("git pull origin main".split())
|
| 53 |
subprocess.call(["git", "add", "."])
|
| 54 |
-
subprocess.call(["git", "commit", "-m", f"
|
| 55 |
subprocess.call(["git", "push"])
|
| 56 |
-
typer.echo("
|
| 57 |
-
|
| 58 |
|
| 59 |
if __name__ == "__main__":
|
| 60 |
app()
|
|
|
|
| 37 |
for task in tasks:
|
| 38 |
downsample_rate = upstream.get_downsample_rates(task)
|
| 39 |
assert isinstance(downsample_rate, int)
|
| 40 |
+
|
|
|
|
| 41 |
except:
|
| 42 |
print("Please check the Upstream Specification on https://superbbenchmark.org/challenge")
|
| 43 |
raise
|
| 44 |
|
| 45 |
typer.echo("All submission files validated!")
|
| 46 |
+
typer.echo("Now you can upload these files to huggingface's Hub.")
|
| 47 |
|
| 48 |
|
| 49 |
@app.command()
|
| 50 |
+
def upload(submission_name: str):
|
| 51 |
subprocess.call("git pull origin main".split())
|
| 52 |
subprocess.call(["git", "add", "."])
|
| 53 |
+
subprocess.call(["git", "commit", "-m", f"Upload Upstream: {submission_name} "])
|
| 54 |
subprocess.call(["git", "push"])
|
| 55 |
+
typer.echo("Upload successful!")
|
| 56 |
+
typer.echo("Now, please go to https://superbbenchmark.org/submit to make a submission.")
|
| 57 |
|
| 58 |
if __name__ == "__main__":
|
| 59 |
app()
|
{{cookiecutter.repo_name}}/expert.py
CHANGED
|
@@ -25,7 +25,7 @@ class Model(nn.Module):
|
|
| 25 |
return [hidden, feature]
|
| 26 |
|
| 27 |
class UpstreamExpert(nn.Module):
|
| 28 |
-
def __init__(self, ckpt: str = "model.pt", **kwargs):
|
| 29 |
"""
|
| 30 |
Args:
|
| 31 |
ckpt:
|
|
@@ -35,7 +35,7 @@ class UpstreamExpert(nn.Module):
|
|
| 35 |
super().__init__()
|
| 36 |
self.name = "[Example UpstreamExpert]"
|
| 37 |
|
| 38 |
-
|
| 39 |
ckpt = torch.load(ckpt, map_location="cpu")
|
| 40 |
self.model = Model()
|
| 41 |
self.model.load_state_dict(ckpt)
|
|
|
|
| 25 |
return [hidden, feature]
|
| 26 |
|
| 27 |
class UpstreamExpert(nn.Module):
|
| 28 |
+
def __init__(self, ckpt: str = "./model.pt", **kwargs):
|
| 29 |
"""
|
| 30 |
Args:
|
| 31 |
ckpt:
|
|
|
|
| 35 |
super().__init__()
|
| 36 |
self.name = "[Example UpstreamExpert]"
|
| 37 |
|
| 38 |
+
# You can use ckpt to load your pretrained weights
|
| 39 |
ckpt = torch.load(ckpt, map_location="cpu")
|
| 40 |
self.model = Model()
|
| 41 |
self.model.load_state_dict(ckpt)
|