Skip to content

Instantly share code, notes, and snippets.

@Tob-iee

Tob-iee/model_to_picsellia.py

Created September 13, 2023 14:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Tob-iee/7cb0f1a9355647c1c7e629cb3b653a74 to your computer and use it in GitHub Desktop.
Save Tob-iee/7cb0f1a9355647c1c7e629cb3b653a74 to your computer and use it in GitHub Desktop.
Download ZIP
Store all the model files in your Picsellia experiment artifact.
Raw
model_to_picsellia.py
# send the trained model to picsellia
model_latest_path = os.path.join(cwd, save_dir)
file_list = os.listdir(model_latest_path)
print(file_list)
for files in file_list:
model_latest_files = os.path.join(model_latest_path, files)
experiment.store(files, model_latest_files)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment