Update prototype_v8_refined_soft_hand_loss.py

prototype_v8_refined_soft_hand_loss.py

@@ -168,12 +168,12 @@ class SVAE(nn.Module):
 
 # -- Training --
 
-def train(epochs=100, lr=1e-3, cv_weight=0.3, boost=0.5,
+def train(epochs=400, lr=3e-4, cv_weight=0.3, boost=0.5,
           sigma=0.15, device='cuda'):
     device = torch.device(device if torch.cuda.is_available() else 'cpu')
-    train_loader, test_loader = get_cifar10(batch_size=256)
+    train_loader, test_loader = get_cifar10(batch_size=512)
 
-    V, D = 40, 24
+    V, D = 256, 24
     target_cv = 0.1250
     model = SVAE(matrix_v=V, D=D).to(device)
     opt = torch.optim.Adam(model.parameters(), lr=lr)