blob: 8708d46a0f97a39e513225e67fc05976468aaa9f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
usage: depth_scaling_benchmark.py [-h]
[--dataset {mnist,fashion_mnist,cifar10,cifar100}]
[--depths DEPTHS [DEPTHS ...]] [--T T]
[--epochs EPOCHS] [--batch_size BATCH_SIZE]
[--lr LR] [--lambda_reg LAMBDA_REG]
[--lambda_target LAMBDA_TARGET]
[--data_dir DATA_DIR] [--out_dir OUT_DIR]
[--device DEVICE] [--seed SEED]
[--no-progress]
[--reg_type {squared,hinge,asymmetric,extreme}]
[--warmup_epochs WARMUP_EPOCHS]
[--stable_init]
[--lyap_threshold LYAP_THRESHOLD]
depth_scaling_benchmark.py: error: argument --reg_type: invalid choice: 'adaptive_sigmoid' (choose from 'squared', 'hinge', 'asymmetric', 'extreme')
|
