-
Notifications
You must be signed in to change notification settings - Fork 20
Expand file tree
/
Copy pathtraining.sh
More file actions
15 lines (14 loc) · 840 Bytes
/
Copy pathtraining.sh
File metadata and controls
15 lines (14 loc) · 840 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/bin/bash
CUDA_VISIBLE_DEVICES=0 python train_v1.py --arch alexnet \
--batch-size 256 \
--epochs 200 \
--lr 0.01 \
--dataset cifar10 \
--norm-type bn \
--key-type shuffle \
--sign-loss 0.1 \
--passport-config passport_configs/alexnet_passport.json \
--lr-config lr_configs/default.json \
--save-interval 0 \
--exp-id 1 \
--tag exptag