s481825/AI_PROJECT
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

neural_network #25

Merged
s481834 merged 12 commits from neural_network into refactor 2024-06-04 13:23:20 +02:00
Conversation 0 Commits 12 Files Changed 14 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 34bfe78218 - Show all commits

2
neuralnetwork.py
View File

@ -14,7 +14,7 @@ fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'su
torch.manual_seed(42)
device = torch.device('cuda')
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
def getTransformation():
transform=transforms.Compose([