YanCote / IFT6268-simclr

Project for IFT6268
0 stars 0 forks source link

Implement hyper-parameter saving when lunching new training loop. #8

Open marued opened 3 years ago

marued commented 3 years ago

Adding code to TensorFlow to support MLFlow. Look at what TensorBoard already provide

sgaut023 commented 3 years ago

@YanCote To save the experiments, we need to do:

import mlflow mlflow.set_experiment(‘NOM DE L’EXPERIENCE’)

to start an experiment

with mlflow.start_run():

log tes params

    mlflow.log_param(‘pca_n’,pca_n_components)
   mlflow.log_metric(‘accuracy’,metric)