Commit 06c041ad authored by Belyaeva Oksana's avatar Belyaeva Oksana
Browse files

added requirenments version

parent b69cab49
......@@ -71,10 +71,7 @@ if __name__ == "__main__":
device = torch.device("cuda:0")
else:
device = torch.device("cpu")
#datas = DatasetExecutorCIFAR()
#loader = datas.load_train_dataset()
#dataiter = iter(loader)
#sample = dataiter.__next__()
net = ClassificationModelTorch()
net.to(device)
net_executor = NetExecutor(net)
......
tensorflow~=2.2.0
tqdm~=4.46.1
numpy~=1.18.5
torch
torchvision
matplotlib
opencv-python
pdf2image
imutils
pandas
scikit-image
\ No newline at end of file
torch==1.5.0
torchvision==0.6.0
matplotlib==3.0.3
opencv-python==4.2.0.34
pdf2image==1.13.1
imutils==0.5.3
pandas==0.25.3
scikit-image==0.15.0
\ No newline at end of file
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment