PyTorch के साथ Deep Learning परिचय
Jasmin Ludolf
Senior Data Science Content Developer, DataCamp
| समस्या | समाधान |
|---|---|
| डेटासेट पर्याप्त बड़ा नहीं है | अधिक डेटा लें / डेटा ऑगमेंटेशन करें |
| मॉडल की क्षमता बहुत अधिक है | मॉडल का आकार घटाएँ / ड्रॉपआउट जोड़ें |
| वेट्स बहुत बड़े हैं | वेट डिके |
रणनीतियाँ:
model = nn.Sequential(nn.Linear(8, 4),
nn.ReLU(),
nn.Dropout(p=0.5))
features = torch.randn((1, 8))
print(model(features))
tensor([[1.4655, 0.0000, 0.0000, 0.8456]], grad_fn=<MulBackward0>)
model.train() और इवैल्युएशन में ड्रॉपआउट बंद करने के लिए model.eval() इस्तेमाल करेंoptimizer = optim.SGD(model.parameters(), lr=0.001, weight_decay=0.0001)
weight_decay पैरामीटर से नियंत्रित, जिसे आमतौर पर छोटा रखते हैं (जैसे 0.0001)
PyTorch के साथ Deep Learning परिचय