From 12faa8f355183cf48a377aacd1d502db2f6f1f67 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 5 Nov 2021 19:28:35 +0100 Subject: Add cross entropy config --- training/conf/criterion/cross_entropy.yaml | 1 + 1 file changed, 1 insertion(+) create mode 100644 training/conf/criterion/cross_entropy.yaml (limited to 'training/conf/criterion/cross_entropy.yaml') diff --git a/training/conf/criterion/cross_entropy.yaml b/training/conf/criterion/cross_entropy.yaml new file mode 100644 index 0000000..b44f215 --- /dev/null +++ b/training/conf/criterion/cross_entropy.yaml @@ -0,0 +1 @@ +_target_: torch.nn.CrossEntropyLoss -- cgit v1.2.3-70-g09d2