Skip to content
Snippets Groups Projects
Commit bc4b7aab authored by Zhihuang Xie's avatar Zhihuang Xie
Browse files

Merge branch 'hop4_hop3_complexity' into 'VTM-11.0_nnvc'

Hop4 hop3 complexity

See merge request jvet-ahg-nnvc/VVCSoftware_VTM!237
parents 1949f421 aa2d9ede
No related branches found
No related tags found
No related merge requests found
......@@ -35,7 +35,7 @@
"D4" : 16,
"D5" : 16,
"D6" : 48,
"N" : 24,
"N" : 23,
"C" : 64,
"C1" : 192,
"C21" : 32,
......
......@@ -170,15 +170,15 @@ class Attn(nn.Module):
self.conv1_1 = nn.Conv2d(c, c, kernel_size=1, bias=False)
self.conv1_2 = nn.Conv2d(c, c, kernel_size=1, bias=False)
self.conv1_3 = nn.Conv2d(c, c, kernel_size=1, bias=False)
group = 8
self.conv3_1 = nn.Conv2d(
c, c, kernel_size=(3, 3), padding=(1, 1), groups=c, bias=False
c, c, kernel_size=(3, 3), padding=(1, 1), groups=group, bias=False
)
self.conv3_2 = nn.Conv2d(
c, c, kernel_size=(3, 3), padding=(1, 1), groups=c, bias=False
c, c, kernel_size=(3, 3), padding=(1, 1), groups=group, bias=False
)
self.conv3_3 = nn.Conv2d(
c, c, kernel_size=(3, 3), padding=(1, 1), groups=c, bias=False
c, c, kernel_size=(3, 3), padding=(1, 1), groups=group, bias=False
)
self.conv2 = nn.Conv2d(c, c, kernel_size=1, bias=False)
......@@ -373,11 +373,12 @@ class SADLNet(nn.Sequential):
patch_size = (
(patch_size, patch_size) if isinstance(patch_size, int) else patch_size
)
global model_for_export
return [
torch.rand(
batch_size, conv.in_channels, *patch_size, device=conv[0].weight.device
)
for conv in self[0].branches
for conv in model_for_export[0].branches
]
def to_onnx(
......
"0 11 1 11 2 11 3 11 4 11 5 11 6 11 7 0 8 11 10 11 12 11 13 0 14 11 16 11 18 11 19 0 20 11 22 11 24 11 25 0 26 11 28 11 30 11 31 0 32 11 34 11 36 11 37 0 38 11 40 11 42 0 44 11 45 0 46 11 48 11 50 11 51 0 52 11 54 11 56 11 57 0 58 11 60 11 62 11 63 0 64 11 66 11 67 0 68 11 70 11 72 0 74 11 75 0 76 11 78 11 79 0 80 11 82 11 83 0 84 11 87 11 88 0 89 11 91 11 93 11 94 0 95 11 97 11 98 0 99 11 101 11 103 0 105 11 106 0 107 11 109 11 110 0 111 11 113 11 114 0 115 11 118 11 119 0 120 11 122 11 124 11 125 0 126 11 128 11 129 0 130 11 132 11 134 0 136 11 137 0 138 11 140 11 141 0 142 11 144 11 145 0 146 11 149 11 150 0 151 11 153 11 155 11 156 0 157 11 159 11 160 0 161 11 163 11 165 0 167 11 168 0 169 11 171 11 172 0 173 11 175 11 176 0 177 11 180 11 181 0 182 11 184 11 186 11 187 0 188 11 190 11 191 0 192 11 194 11 196 0 198 11 199 0 200 11 202 11 203 0 204 11 206 11 207 0 208 11 211 11 212 0 213 11 215 11 217 11 218 0 219 11 221 11 222 0 223 11 225 11 227 0 229 11 230 0 231 11 233 11 234 0 235 11 237 11 238 0 239 11 242 11 243 0 244 11 246 11 248 11 249 0 250 11 252 11 253 0 254 11 256 11 258 0 260 11 261 0 262 11 264 11 265 0 266 11 268 11 269 0 270 11 273 11 274 0 275 11 277 11 279 11 280 0 281 11 283 11 284 0 285 11 287 11 289 0 291 11 292 0 293 11 295 11 296 0 297 11 299 11 300 0 301 11 303 11 304 0 305 11 306 0 307 11 308 0 309 11 310 0 311 11 312 0 313 11 314 0 316 0 318 0 320 0 322 0 324 0 326 0 328 0 330 0 332 0 334 0 336 0 338 0 340 0 342 0 344 0 345 0 347 0 349 0 351 0 352 0 354 0 357 11 358 0 361 11 362 0 363 11 365 11 367 11 368 0 369 11 371 11 372 0 373 11 375 11 377 0 379 11 380 0 381 11 383 11 384 0 385 11 387 11 388 0 389 11 392 11 393 0 394 11 396 11 398 11 399 0 400 11 402 11 403 0 404 11 406 11 408 0 410 11 411 0 412 11 414 11 415 0 416 11 418 11 419 0 420 11 423 11 424 0 425 11 427 11 429 11 430 0 431 11 433 11 434 0 435 11 437 11 439 0 441 11 442 0 443 11 445 11 446 0 447 11 449 11 450 0 451 11 454 11 455 0 456 11 458 11 460 11 461 0 462 11 464 11 465 0 466 11 468 11 470 0 472 11 473 0 474 11 476 11 477 0 478 11 480 11 481 0 482 11 485 11 486 0 487 11 489 11 491 11 492 0 493 11 495 11 496 0 497 11 499 11 501 0 503 11 504 0 505 11 507 11 508 0 509 11 511 11 512 0 513 11 516 11 517 0 518 11 520 11 522 11 523 0 524 11 526 11 527 0 528 11 530 11 532 0 534 11 535 0 536 11 538 11 539 0 540 11 542 11 543 0 544 11 547 11 548 0 549 11 551 11 553 11 554 0 555 11 557 11 558 0 559 11 561 11 563 0 565 11 566 0 567 11 569 11 570 0 571 11 573 11 574 0 575 11 577 11 578 0 579 11 580 0 581 11 582 0 583 11 584 0 585 11 586 0 587 11 588 0 590 0 592 0 594 0 596 0 598 0 600 0 602 0 604 0 606 0 608 0 610 0 612 0 614 0 616 0 618 0 619 0 621 0 623 0 625 0 626 0 628 0 631 11 632 0 635 11 636 0 637 11 639 11 641 11 642 0 643 11 645 11 646 0 647 11 649 11 651 0 653 11 654 0 655 11 657 11 658 0 659 11 661 11 662 0 663 11 666 11 667 0 668 11 670 11 672 11 673 0 674 11 676 11 677 0 678 11 680 11 682 0 684 11 685 0 686 11 688 11 689 0 690 11 692 11 693 0 694 11 697 11 698 0 699 11 701 11 703 11 704 0 705 11 707 11 708 0 709 11 711 11 713 0 715 11 716 0 717 11 719 11 720 0 721 11 723 11 724 0 725 11 728 11 729 0 730 11 732 11 734 11 735 0 736 11 738 11 739 0 740 11 742 11 744 0 746 11 747 0 748 11 750 11 751 0 752 11 754 11 755 0 756 11 759 11 760 0 761 11 763 11 765 11 766 0 767 11 769 11 770 0 771 11 773 11 775 0 777 11 778 0 779 11 781 11 782 0 783 11 785 11 786 0 787 11 790 11 791 0 792 11 794 11 796 11 797 0 798 11 800 11 801 0 802 11 804 11 806 0 808 11 809 0 810 11 812 11 813 0 814 11 816 11 817 0 818 11 821 11 822 0 823 11 825 11 827 11 828 0 829 11 831 11 832 0 833 11 835 11 837 0 839 11 840 0 841 11 843 11 844 0 845 11 847 11 848 0 849 11 852 11 853 0 854 11 856 11 858 11 859 0 860 11 862 11 863 0 864 11 866 11 868 0 870 11 871 0 872 11 874 11 875 0 876 11 878 11 879 0 880 11 883 11 884 0 885 11 887 11 889 11 890 0 891 11 893 11 894 0 895 11 897 11 899 0 901 11 902 0 903 11 905 11 906 0 907 11 909 11 910 0 911 11 914 11 915 0 916 11 918 11 920 11 921 0 922 11"
\ No newline at end of file
0 11 1 11 2 11 3 11 4 11 5 11 6 0 7 0 8 0 10 0 12 0 13 0 14 0 16 0 18 0 19 0 20 0 22 0 24 0 25 0 26 0 28 0 30 0 31 0 32 0 34 0 36 0 37 0 38 0 40 0 42 0 44 0 45 0 46 0 48 0 50 0 51 0 52 0 54 0 56 0 57 0 58 0 60 0 62 0 63 0 64 0 66 0 67 0 68 0 70 0 72 0 74 0 75 0 76 0 78 0 79 0 80 0 82 0 83 0 84 0 87 0 88 0 89 0 91 0 93 0 94 0 95 0 97 0 98 0 99 0 101 0 103 0 105 0 106 0 107 0 109 0 110 0 111 0 113 0 114 0 115 0 118 0 119 0 120 0 122 0 124 0 125 0 126 0 128 0 129 0 130 0 132 0 134 0 136 0 137 0 138 0 140 0 141 0 142 0 144 0 145 0 146 0 149 0 150 0 151 0 153 0 155 0 156 0 157 0 159 0 160 0 161 0 163 0 165 0 167 0 168 0 169 0 171 0 172 0 173 0 175 0 176 0 177 0 180 0 181 0 182 0 184 0 186 0 187 0 188 0 190 0 191 0 192 0 194 0 196 0 198 0 199 0 200 0 202 0 203 0 204 0 206 0 207 0 208 0 211 0 212 0 213 0 215 0 217 0 218 0 219 0 221 0 222 0 223 0 225 0 227 0 229 0 230 0 231 0 233 0 234 0 235 0 237 0 238 0 239 0 242 0 243 0 244 0 246 0 248 0 249 0 250 0 252 0 253 0 254 0 256 0 258 0 260 0 261 0 262 0 264 0 265 0 266 0 268 0 269 0 270 0 273 0 274 0 275 0 277 0 279 0 280 0 281 0 283 0 284 0 285 0 287 0 289 0 291 0 292 0 293 0 295 0 296 0 297 0 299 0 300 0 301 0 303 0 304 0 305 0 306 0 307 0 308 0 309 0 310 0 311 0 312 0 313 0 314 0 316 11 318 0 320 11 322 11 324 0 326 11 328 11 330 0 332 11 334 11 336 11 338 11 340 11 342 11 344 0 345 11 347 11 349 11 351 0 352 11 354 11 357 0 358 0 361 0 362 0 363 0 365 0 367 0 368 0 369 0 371 0 372 0 373 0 375 0 377 0 379 0 380 0 381 0 383 0 384 0 385 0 387 0 388 0 389 0 392 0 393 0 394 0 396 0 398 0 399 0 400 0 402 0 403 0 404 0 406 0 408 0 410 0 411 0 412 0 414 0 415 0 416 0 418 0 419 0 420 0 423 0 424 0 425 0 427 0 429 0 430 0 431 0 433 0 434 0 435 0 437 0 439 0 441 0 442 0 443 0 445 0 446 0 447 0 449 0 450 0 451 0 454 0 455 0 456 0 458 0 460 0 461 0 462 0 464 0 465 0 466 0 468 0 470 0 472 0 473 0 474 0 476 0 477 0 478 0 480 0 481 0 482 0 485 0 486 0 487 0 489 0 491 0 492 0 493 0 495 0 496 0 497 0 499 0 501 0 503 0 504 0 505 0 507 0 508 0 509 0 511 0 512 0 513 0 516 0 517 0 518 0 520 0 522 0 523 0 524 0 526 0 527 0 528 0 530 0 532 0 534 0 535 0 536 0 538 0 539 0 540 0 542 0 543 0 544 0 547 0 548 0 549 0 551 0 553 0 554 0 555 0 557 0 558 0 559 0 561 0 563 0 565 0 566 0 567 0 569 0 570 0 571 0 573 0 574 0 575 0 577 0 578 0 579 0 580 0 581 0 582 0 583 0 584 0 585 0 586 0 587 0 588 0 590 11 592 0 594 11 596 11 598 0 600 11 602 11 604 0 606 11 608 11 610 11 612 11 614 11 616 11 618 0 619 11 621 11 623 11 625 0 626 11 628 11 631 0 632 0 635 0 636 0 637 0 639 0 641 0 642 0 643 0 645 0 646 0 647 0 649 0 651 0 653 0 654 0 655 0 657 0 658 0 659 0 661 0 662 0 663 0 666 0 667 0 668 0 670 0 672 0 673 0 674 0 676 0 677 0 678 0 680 0 682 0 684 0 685 0 686 0 688 0 689 0 690 0 692 0 693 0 694 0 697 0 698 0 699 0 701 0 703 0 704 0 705 0 707 0 708 0 709 0 711 0 713 0 715 0 716 0 717 0 719 0 720 0 721 0 723 0 724 0 725 0 728 0 729 0 730 0 732 0 734 0 735 0 736 0 738 0 739 0 740 0 742 0 744 0 746 0 747 0 748 0 750 0 751 0 752 0 754 0 755 0 756 0 759 0 760 0 761 0 763 0 765 0 766 0 767 0 769 0 770 0 771 0 773 0 775 0 777 0 778 0 779 0 781 0 782 0 783 0 785 0 786 0 787 0 790 0 791 0 792 0 794 0 796 0 797 0 798 0 800 0 801 0 802 0 804 0 806 0 808 0 809 0 810 0 812 0 813 0 814 0 816 0 817 0 818 0 821 0 822 0 823 0 825 0 827 0 828 0 829 0 831 0 832 0 833 0 835 0 837 0 839 0 840 0 841 0 843 0 844 0 845 0 847 0 848 0 849 0 852 0 853 0 854 0 856 0 858 0 859 0 860 0 862 0 863 0 864 0 866 0 868 0 870 0 871 0 872 0 874 0 875 0 876 0 878 0 879 0 880 0 883 0 884 0 885 0 887 0 889 0 890 0 891 0
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment