Commit 69590f42 authored by Carl De Sousa Trias's avatar Carl De Sousa Trias
Browse files

Some updates for Reference Software 1.2

parent 0599c34c
......@@ -44,8 +44,8 @@ if __name__ == '__main__':
model.classifier = nn.Linear(25088, 10)
model.to(device)
# watermarking section (change here to test another method) #######################################
tools = Adi_tools()
reload = 'vgg16_Adi'
tools = Uchi_tools()
reload = 'Resources/vgg16_Uchi'
watermarking_dict = np.load(reload+'_watermarking_dict.npy', allow_pickle=True).item()
# watermarking section (END change here to test another method) ###################################
name = '_quantization'
......
Supports Markdown
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