Slightly changes

loss_function/own_loss_function.py

@@ -318,7 +318,7 @@ my_nn.eval()
 opt = optim.SGD(params=my_nn.parameters(), lr=0.1)
 result = my_nn(weightmatrix.view(batch_size,1, 1,100))
 
-for epoch in tqdm(range(15)):
+for epoch in tqdm(range(15000)):
     result = my_nn(weightmatrix.view(batch_size,1,1, 100))
     loss = criterion(result_given=result.view(batch_size,1,10,10), points_given=points)
     print("loss" , loss)

loss_function/own_loss_function_cv2.py

@@ -229,7 +229,7 @@ result = my_nn(random_data.view(batch_size,1, 1,100))
 
 print(result.view(10,10))
 
-for epoch in tqdm(range(75)):
+for epoch in tqdm(range(15000)):
     result = my_nn(random_data.view(batch_size,1,1, 100))
 
     start_loss, gap_loss, cluster_size_loss = criterion(result_given=result.view(batch_size,1,10,10), points_given=points)