aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat image2sketch 变更模型

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-09-30 11:15:43 +08:00
parent 2d9b74a398
commit 5a5bb07f3b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/image2sketch_2/server.py
View File

@@ -92,7 +92,7 @@ class Generator(nn.Module):
model1 = Generator(3, 1, 3)
model1.load_state_dict(torch.load('service/image2sketch_2/model.pth', map_location=torch.device('cpu')))
model1.load_state_dict(torch.load('app/service/image2sketch_2/model.pth', map_location=torch.device('cpu')))
model1.eval()