Browse Source

更新 'ab.py'

master
1073270530 2 months ago
parent
commit
053c3a247f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ab.py

+ 1
- 1
ab.py View File

@@ -404,7 +404,7 @@ def launch():
args.num_classes = 12*11
args.dataset_name = "faces"
args.device = torch.device("cuda:0" if torch.cuda.is_available() else 'cpu')
args.lr = 1e-4
args.lr = 1e-5
train(args)



Loading…
Cancel
Save