thanks after correction now getting weird error. I cannot understand
warnings.warn(warning.format(ret))
[TRAIN] Epoch 1/3:   0%|                        | 1/250 [00:00<02:24,  1.73it/s]
Traceback (most recent call last):
File “main.py”, line 58, in 
main(args, CORE)
File “main.py”, line 48, in main
CORE.train(args)
File “/home/ali/BioNet_project/pytorch_version/utils.py”, line 89, in train
l.backward()
File “/home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/tensor.py”, line 198, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File “/home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/autograd/init.py”, line 100, in backward
allow_unreachable=True)  # allow_unreachable flag
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemv(handle, op, m, n, &alpha, a, lda, x, incx, &beta, y, incy) (gemv at /opt/conda/conda-bld/pytorch_1587428091666/work/aten/src/ATen/cuda/CUDABlas.cpp:318)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x4e (0x7fc15acaab5e in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libc10.so)
frame #1:  + 0xdb9fa7 (0x7fc15bc90fa7 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cuda.so)
frame #2: at::native::(anonymous namespace)::slow_conv_transpose2d_acc_grad_parameters_cuda_template(at::Tensor const&, at::Tensor const&, at::Tensor&, at::Tensor&, at::Tensor const&, at::Tensor const&, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, int) + 0xea6 (0x7fc15d583e86 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cuda.so)
frame #3: at::native::slow_conv_transpose2d_backward_cuda(at::Tensor const&, at::Tensor const&, at::Tensor const&, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, c10::ArrayRef, at::Tensor const&, at::Tensor const&, std::array<bool, 3ul>) + 0x323 (0x7fc15d588c93 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cuda.so)
frame #4:  + 0xe1f64d (0x7fc15bcf664d in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cuda.so)
frame #5:  + 0xe28007 (0x7fc15bcff007 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cuda.so)
frame #6:  + 0x29e286e (0x7fc18486e86e in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #7:  + 0xe23c87 (0x7fc182cafc87 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #8: torch::autograd::generated::SlowConvTranspose2DBackward::apply(std::vector<at::Tensor, std::allocatorat::Tensor >&&) + 0x516 (0x7fc1844b7c46 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #9:  + 0x2ae8215 (0x7fc184974215 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #10: torch::autograd::Engine::evaluate_function(std::shared_ptrtorch::autograd::GraphTask&, torch::autograd::Node*, torch::autograd::InputBuffer&) + 0x16f3 (0x7fc184971513 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #11: torch::autograd::Engine::thread_main(std::shared_ptrtorch::autograd::GraphTask const&, bool) + 0x3d2 (0x7fc1849722f2 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #12: torch::autograd::Engine::thread_init(int) + 0x39 (0x7fc18496a969 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_cpu.so)
frame #13: torch::autograd::python::PythonEngine::thread_init(int) + 0x38 (0x7fc187cb0c38 in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/libtorch_python.so)
frame #14:  + 0xc819d (0x7fc18a72b19d in /home/ali/anaconda3/envs/torch/lib/python3.6/site-packages/torch/lib/…/…/…/…/./libstdc++.so.6)
frame #15:  + 0x9609 (0x7fc1a3348609 in /lib/x86_64-linux-gnu/libpthread.so.0)
frame #16: clone + 0x43 (0x7fc1a326f293 in /lib/x86_64-linux-gnu/libc.so.6)