Commit 137fb48b authored by mashun1's avatar mashun1
Browse files

Update sample_conditional.py

parent f56ae884
...@@ -288,7 +288,7 @@ def get_data(config): ...@@ -288,7 +288,7 @@ def get_data(config):
return data return data
@st.cache(allow_output_mutation=True, suppress_st_warning=True) # @st.cache(allow_output_mutation=True, suppress_st_warning=True)
def load_model_and_dset(config, ckpt, gpu, eval_mode): def load_model_and_dset(config, ckpt, gpu, eval_mode):
# get data # get data
dsets = get_data(config) # calls data.config ... dsets = get_data(config) # calls data.config ...
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment