This commit is contained in:
leo
2024-03-07 13:49:41 +08:00
parent 4b38e5a854
commit 034ad2a3f1
2 changed files with 22 additions and 16 deletions

BIN
data.db

Binary file not shown.

38
main.py
View File

@@ -84,23 +84,29 @@ with data_frame_container:
'is_validation': "是否是验证集", 'is_validation': "是否是验证集",
}) })
col1, col2, col3, col4, col5 = st.columns(5)
with config_container: with config_container:
st.json(st.session_state.configs)
with st.sidebar:
st.divider() st.divider()
st.slider(label='evolve_r', key='evolve_r',
min_value=0.0, max_value=0.5, step=0.01, with col1:
on_change=update_config) st.slider(label='evolve_r', key='evolve_r',
st.selectbox(label='n_trail', key='n_trail', min_value=0.0, max_value=0.5, step=0.01,
options=(i for i in range(10, 51, 10))) on_change=update_config)
st.selectbox(label='n_epoch', key='n_epoch', with col2:
options=(i for i in range(1, 6))) st.selectbox(label='n_trail', key='n_trail',
st.selectbox(label='ckpt_path', key='ckpt_path', options=(i for i in range(10, 51, 10)))
options=(os.listdir(BASE_CKPT_DIR))) st.selectbox(label='n_epoch', key='n_epoch',
st.selectbox(label='mode', key='mode', options=(i for i in range(1, 6)))
format_func=lambda x: RUN_MODE[x], with col3:
options=RUN_MODE, st.selectbox(label='ckpt_path', key='ckpt_path',
on_change=update_config) options=(os.listdir(BASE_CKPT_DIR)))
with col4:
st.selectbox(label='mode', key='mode',
format_func=lambda x: RUN_MODE[x],
options=RUN_MODE,
on_change=update_config)
with col5:
st.json(st.session_state.configs)
st.divider() st.divider()
st.button("启动", on_click=train) st.button("启动", on_click=train)