/PTYNet

The official implementation of the paper "Free Fine-tuning: A Plug-and-Play Watermarking Scheme for Deep Neural Networks".

Primary LanguagePythonMIT LicenseMIT

PTYNet

This repository includes official PyTorch implementation of the paper "Free Fine-tuning: A Plug-and-Play Watermarking Scheme for Deep Neural Networks". [arXiv]

Get Started

We recommend using conda for installation.

conda create -n ptynet python=3.7
conda activate ptynet

After activating conda environment, please install the dependencies running the following commands

pip install -r requirements.txt

Our models and datasets will be available soon.