UGATIT_with_torch.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"name":"UGATIT_with_torch.ipynb","provenance":[],"authorship_tag":"ABX9TyNPpdNvZIqW51GDVZZbksst"},"kernelspec":{"name":"python3","display_name":"Python 3"},"accelerator":"GPU"},"cells":[{"cell_type":"code","metadata":{"id":"-uvN12sSnYUX","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":122},"outputId":"8703211a-1436-4cd2-a0f6-09091eb7b9b2","executionInfo":{"status":"ok","timestamp":1586246954095,"user_tz":-330,"elapsed":19770,"user":{"displayName":"talha khalid","photoUrl":"","userId":"01979772479587971642"}}},"source":["from google.colab import drive\n","drive.mount('/content/drive')"],"execution_count":1,"outputs":[{"output_type":"stream","text":["Go to this URL in a browser: https://accounts.google.com/o/oauth2/auth?client_id=947318989803-6bn6qk8qdgf4n4g3pfee6491hc0brc4i.apps.googleusercontent.com&redirect_uri=urn%3aietf%3awg%3aoauth%3a2.0%3aoob&response_type=code&scope=email%20https%3a%2f%2fwww.googleapis.com%2fauth%2fdocs.test%20https%3a%2f%2fwww.googleapis.com%2fauth%2fdrive%20https%3a%2f%2fwww.googleapis.com%2fauth%2fdrive.photos.readonly%20https%3a%2f%2fwww.googleapis.com%2fauth%2fpeopleapi.readonly\n","\n","Enter your authorization code:\n","··········\n","Mounted at /content/drive\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"F5wNYk_FnJPc","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":34},"outputId":"b4be9f7d-b058-42af-d33e-6ce3b6bb3966","executionInfo":{"status":"ok","timestamp":1586246967876,"user_tz":-330,"elapsed":1022,"user":{"displayName":"talha khalid","photoUrl":"","userId":"01979772479587971642"}}},"source":["cd /content/drive/My Drive/a_sharma/UGATIT_pytorch"],"execution_count":4,"outputs":[{"output_type":"stream","text":["/content/drive/My Drive/a_sharma/UGATIT_pytorch\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"ReHz4DJCnjlc","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":85},"outputId":"98313d5c-efec-46f8-d07f-16e901354b5d","executionInfo":{"status":"ok","timestamp":1586245855600,"user_tz":-330,"elapsed":5451,"user":{"displayName":"talha khalid","photoUrl":"","userId":"01979772479587971642"}}},"source":["!git clone https://github.com/znxlwm/UGATIT-pytorch.git"],"execution_count":3,"outputs":[{"output_type":"stream","text":["Cloning into 'UGATIT-pytorch'...\n","remote: Enumerating objects: 70, done.\u001b[K\n","remote: Total 70 (delta 0), reused 0 (delta 0), pack-reused 70\u001b[K\n","Unpacking objects: 100% (70/70), done.\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"BnVuPYWfntPu","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":34},"outputId":"1a51e277-a1b9-44f5-a5c2-3c7563a7c587","executionInfo":{"status":"ok","timestamp":1586246969372,"user_tz":-330,"elapsed":653,"user":{"displayName":"talha khalid","photoUrl":"","userId":"01979772479587971642"}}},"source":["cd UGATIT-pytorch/"],"execution_count":5,"outputs":[{"output_type":"stream","text":["/content/drive/My Drive/a_sharma/UGATIT_pytorch/UGATIT-pytorch\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"Fh1tRde0nxJ3","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":564},"outputId":"78f37b2f-88aa-4fd7-a8b2-477888bcc72b","executionInfo":{"status":"ok","timestamp":1586247703035,"user_tz":-330,"elapsed":13627,"user":{"displayName":"talha khalid","photoUrl":"","userId":"01979772479587971642"}}},"source":["!python main.py --dataset face2anime"],"execution_count":7,"outputs":[{"output_type":"stream","text":["number of epochs must be larger than or equal to one\n","\n","##### Information #####\n","# light :  False\n","# dataset :  face2anime\n","# batch_size :  1\n","# iteration per epoch :  1000000\n","\n","##### Generator #####\n","# residual blocks :  4\n","\n","##### Discriminator #####\n","# discriminator layer :  6\n","\n","##### Weight #####\n","# adv_weight :  1\n","# cycle_weight :  10\n","# identity_weight :  10\n","# cam_weight :  1000\n","tcmalloc: large alloc 1073741824 bytes == 0x50a6000 @  0x7f9f7b591b6b 0x7f9f7b5b1379 0x7f9f24d0db4a 0x7f9f24d0f5fa 0x7f9f2703f78a 0x7f9f2728830b 0x7f9f272cfb37 0x7f9f70156ad5 0x7f9f7015717b 0x7f9f7015b160 0x7f9f6ffa8496 0x551b15 0x5aa6ec 0x50abb3 0x50c5b9 0x508245 0x509642 0x595311 0x54a6ff 0x551b81 0x5aa6ec 0x50abb3 0x50d390 0x508245 0x509642 0x595311 0x54a6ff 0x551b81 0x5aa6ec 0x50abb3 0x50d390\n","training start !\n","Traceback (most recent call last):\n","  File \"main.py\", line 83, in <module>\n","    main()\n","  File \"main.py\", line 75, in main\n","    gan.train()\n","  File \"/content/drive/My Drive/a_sharma/UGATIT_pytorch/UGATIT-pytorch/UGATIT.py\", line 196, in train\n","    self.D_optim.step()\n","  File \"/usr/local/lib/python3.6/dist-packages/torch/optim/adam.py\", line 103, in step\n","    denom = (exp_avg_sq.sqrt() / math.sqrt(bias_correction2)).add_(group['eps'])\n","RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 7.43 GiB total capacity; 6.28 GiB already allocated; 70.94 MiB free; 6.84 GiB reserved in total by PyTorch)\n"],"name":"stdout"}]},{"cell_type":"code","metadata":{"id":"IY2YXDYcrgVM","colab_type":"code","colab":{}},"source":[""],"execution_count":0,"outputs":[]}]}