Hi, all,
    I implement my code about an Autoendocer with dense blcok on the basis of Pytorch.  The total number of parameters around 0.8M.  The image size is 100*100.  the batchsize for training 100. However, when I launch CUDA, an error "cuda out of memory" will be ocurred.  Can anyone can help me. My GPU card : 

-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.14       Driver Version: 430.14       CUDA Version: 10.2     |
|-------------------------------±---------------------±---------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 745     Off  | 00000000:01:00.0  On |                  N/A |
| 25%   59C    P0    N/A /  N/A |   4037MiB /  4043MiB |      1%      Default |
±------------------------------±---------------------±---------------------+
±----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|===========================================================================
Thanks.