Skip to content
Open
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion experiments/classification/cifar10/configs/resnet.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/cifar100/configs/resnet.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 10
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
# ClassificationRoutine
model:
# BatchEnsemble
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.classification.bayesian_lenet
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.CheckpointCollector
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.CheckpointCollector
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.classification.lenet
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
# ClassificationRoutine
model:
# DeepEnsemble
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/mnist/configs/lenet/ema.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.EMA
init_args:
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/mnist/configs/lenet/sghmc.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ trainer:
patience: 1000
check_finite: true
gradient_clip_val: 5
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.CheckpointCollector
init_args:
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/mnist/configs/lenet/sgld.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.CheckpointCollector
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.classification.lenet
init_args:
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/mnist/configs/lenet/swa.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.SWA
init_args:
Expand Down
2 changes: 1 addition & 1 deletion experiments/classification/mnist/configs/lenet/swag.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.SWAG
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
model:
class_path: torch_uncertainty.models.wrappers.Zero
init_args:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ trainer:
monitor: val/cls/Acc
patience: 1000
check_finite: true
model:
routine:
num_classes: 100
in_channels: 3
loss: CrossEntropyLoss
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ trainer:
default_hp_metric: false
callbacks:
- class_path: torch_uncertainty.callbacks.TUClsCheckpoint
model:
routine:
model:
class_path: torch_uncertainty.models.classification.batched_inception_time
init_args:
Expand Down
Loading