search

ksk0629 / quanvolutional_neural_network

This is the repository for one implementation of the algorithm in scientific paper, Quanvolutional Neural Networks: Powering Image Recognition with Quantum Circuits.
MIT License
1 stars 1 forks source link

Add mlflow #9

Closed ksk0629 closed 1 month ago

ksk0629 commented 1 month ago

What to do

Add mlflow and record the training process.

Time

ksk0629 commented 1 month ago

What's left

Add a final model file and preprocessed data, which is an output of QuanvLayer, as artefacts.

ksk0629 commented 1 month ago

When the model is saved, mlflow shows the following warning message. WARNING mlflow.models.model: Input example should be provided to infer model signature if the model signature is not provided when logging the model.

However, it seems like torch.Tensor is not supported yet Ref. So, I will ignore the warning for now.