wolves/AI-learning
1
0
Fork 0
Code Releases Activity

241005

Browse Source
This commit is contained in:
Wolves
2024-10-05 22:43:47 +08:00
parent 176e643f20
commit 804b63b2a3
37 changed files with 5317 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
linear regression/1_cuda.py
View File

@@ -39,7 +39,7 @@ def gradient_descent_runner(points, starting_b, starting_w, learningRate, num_it
def run():
points_np = np.genfromtxt("data1.csv", delimiter=',').astype(np.float32)
points = torch.tensor(points_np, device='cuda:5')
points = torch.tensor(points_np, device='cuda')
learning_rate = 0.0001
initial_b = 0.0
initial_w = 0.0