add slider and link
Browse files
app.py
CHANGED
@@ -22,21 +22,18 @@ sb = st.sidebar
|
|
22 |
sb.title("Train your model")
|
23 |
# wandb_token = sb.text_input("paste your wandb Api key if you want: https://wandb.ai/authorize", type="password")
|
24 |
|
25 |
-
|
26 |
# wandb.login(key=wandb_token)
|
27 |
wandb.login(anonymous="must")
|
28 |
api = wandb.Api()
|
29 |
|
30 |
st.success(f"You should see a new run named **{run_name}**, it\'ll have a green circle while it\'s still active")
|
31 |
|
32 |
-
|
33 |
# render wandb dashboard
|
34 |
components.html(get_project(api, project, entity), height=HEIGHT)
|
35 |
|
36 |
# run params
|
37 |
runs = 1
|
38 |
-
epochs = sb.
|
39 |
-
|
40 |
|
41 |
|
42 |
pseudo_code = """
|
@@ -53,10 +50,13 @@ for i in range(epochs):
|
|
53 |
|
54 |
sb.write(pseudo_code)
|
55 |
|
|
|
56 |
# train model
|
57 |
if sb.button("Run Example"):
|
58 |
|
59 |
print("Running training")
|
60 |
for i in range(runs):
|
61 |
my_bar = sb.progress(0)
|
62 |
-
train(name=run_name, project=project, entity=entity, epochs=epochs, bar=my_bar)
|
|
|
|
|
|
22 |
sb.title("Train your model")
|
23 |
# wandb_token = sb.text_input("paste your wandb Api key if you want: https://wandb.ai/authorize", type="password")
|
24 |
|
|
|
25 |
# wandb.login(key=wandb_token)
|
26 |
wandb.login(anonymous="must")
|
27 |
api = wandb.Api()
|
28 |
|
29 |
st.success(f"You should see a new run named **{run_name}**, it\'ll have a green circle while it\'s still active")
|
30 |
|
|
|
31 |
# render wandb dashboard
|
32 |
components.html(get_project(api, project, entity), height=HEIGHT)
|
33 |
|
34 |
# run params
|
35 |
runs = 1
|
36 |
+
epochs = sb.slider('Number of epochs:', min_value=100, max_value=500, value=100)
|
|
|
37 |
|
38 |
|
39 |
pseudo_code = """
|
|
|
50 |
|
51 |
sb.write(pseudo_code)
|
52 |
|
53 |
+
sb.write("Click π to start logging")
|
54 |
# train model
|
55 |
if sb.button("Run Example"):
|
56 |
|
57 |
print("Running training")
|
58 |
for i in range(runs):
|
59 |
my_bar = sb.progress(0)
|
60 |
+
train(name=run_name, project=project, entity=entity, epochs=epochs, bar=my_bar)
|
61 |
+
|
62 |
+
st.subheader("Check our π₯ [Pytorch Intro colab](https://wandb.me/intro) π₯")
|