|
18 | 18 | }, |
19 | 19 | { |
20 | 20 | "cell_type": "code", |
21 | | - "execution_count": null, |
| 21 | + "execution_count": 1, |
22 | 22 | "metadata": {}, |
23 | 23 | "outputs": [], |
24 | 24 | "source": [ |
|
40 | 40 | }, |
41 | 41 | { |
42 | 42 | "cell_type": "code", |
43 | | - "execution_count": null, |
| 43 | + "execution_count": 2, |
44 | 44 | "metadata": {}, |
45 | 45 | "outputs": [], |
46 | 46 | "source": [ |
|
49 | 49 | }, |
50 | 50 | { |
51 | 51 | "cell_type": "code", |
52 | | - "execution_count": null, |
| 52 | + "execution_count": 3, |
53 | 53 | "metadata": {}, |
54 | 54 | "outputs": [ |
55 | 55 | { |
56 | 56 | "data": { |
57 | 57 | "text/plain": [ |
58 | 58 | "MXResNet constructor\n", |
59 | 59 | " c_in: 3, c_out: 1000\n", |
60 | | - " expansion: 1, groups: 1, dw: False\n", |
| 60 | + " expansion: 1, groups: 1, dw: False, div_groups: None\n", |
61 | 61 | " sa: False, se: False\n", |
62 | | - " stem sizes: [3, 32, 64, 64], stide on 0\n", |
| 62 | + " stem sizes: [3, 32, 64, 64], stride on 0\n", |
63 | 63 | " body sizes [64, 128, 256, 512]\n", |
64 | 64 | " layers: [2, 2, 2, 2]" |
65 | 65 | ] |
66 | 66 | }, |
67 | | - "execution_count": null, |
| 67 | + "execution_count": 3, |
68 | 68 | "metadata": {}, |
69 | 69 | "output_type": "execute_result" |
70 | 70 | } |
|
75 | 75 | }, |
76 | 76 | { |
77 | 77 | "cell_type": "code", |
78 | | - "execution_count": null, |
| 78 | + "execution_count": 4, |
79 | 79 | "metadata": {}, |
80 | 80 | "outputs": [ |
81 | 81 | { |
|
84 | 84 | "([64, 64, 128, 256, 512], [2, 2, 2, 2])" |
85 | 85 | ] |
86 | 86 | }, |
87 | | - "execution_count": null, |
| 87 | + "execution_count": 4, |
88 | 88 | "metadata": {}, |
89 | 89 | "output_type": "execute_result" |
90 | 90 | } |
|
95 | 95 | }, |
96 | 96 | { |
97 | 97 | "cell_type": "code", |
98 | | - "execution_count": null, |
| 98 | + "execution_count": 5, |
99 | 99 | "metadata": {}, |
100 | 100 | "outputs": [ |
101 | 101 | { |
|
121 | 121 | ")" |
122 | 122 | ] |
123 | 123 | }, |
124 | | - "execution_count": null, |
| 124 | + "execution_count": 5, |
125 | 125 | "metadata": {}, |
126 | 126 | "output_type": "execute_result" |
127 | 127 | } |
|
133 | 133 | }, |
134 | 134 | { |
135 | 135 | "cell_type": "code", |
136 | | - "execution_count": null, |
137 | | - "metadata": {}, |
138 | | - "outputs": [ |
139 | | - { |
140 | | - "name": "stdout", |
141 | | - "output_type": "stream", |
142 | | - "text": [ |
143 | | - "torch.Size([16, 64, 32, 32])\n" |
144 | | - ] |
145 | | - } |
146 | | - ], |
147 | | - "source": [ |
148 | | - "#hide\n", |
149 | | - "bs_test = 16\n", |
150 | | - "xb = torch.randn(bs_test, 3, 128, 128)\n", |
151 | | - "y = mxresnet.stem(xb)\n", |
152 | | - "print(y.shape)\n", |
153 | | - "assert y.shape == torch.Size([bs_test, 64, 32, 32]), f\"size\"" |
154 | | - ] |
155 | | - }, |
156 | | - { |
157 | | - "cell_type": "code", |
158 | | - "execution_count": null, |
| 136 | + "execution_count": 6, |
159 | 137 | "metadata": {}, |
160 | 138 | "outputs": [ |
161 | 139 | { |
|
300 | 278 | ")" |
301 | 279 | ] |
302 | 280 | }, |
303 | | - "execution_count": null, |
| 281 | + "execution_count": 6, |
304 | 282 | "metadata": {}, |
305 | 283 | "output_type": "execute_result" |
306 | 284 | } |
|
312 | 290 | }, |
313 | 291 | { |
314 | 292 | "cell_type": "code", |
315 | | - "execution_count": null, |
| 293 | + "execution_count": 7, |
316 | 294 | "metadata": {}, |
317 | 295 | "outputs": [ |
318 | 296 | { |
|
457 | 435 | ")" |
458 | 436 | ] |
459 | 437 | }, |
460 | | - "execution_count": null, |
| 438 | + "execution_count": 7, |
461 | 439 | "metadata": {}, |
462 | 440 | "output_type": "execute_result" |
463 | 441 | } |
|
469 | 447 | }, |
470 | 448 | { |
471 | 449 | "cell_type": "code", |
472 | | - "execution_count": null, |
| 450 | + "execution_count": 8, |
473 | 451 | "metadata": {}, |
474 | 452 | "outputs": [ |
475 | 453 | { |
|
482 | 460 | ")" |
483 | 461 | ] |
484 | 462 | }, |
485 | | - "execution_count": null, |
| 463 | + "execution_count": 8, |
486 | 464 | "metadata": {}, |
487 | 465 | "output_type": "execute_result" |
488 | 466 | } |
|
494 | 472 | }, |
495 | 473 | { |
496 | 474 | "cell_type": "code", |
497 | | - "execution_count": null, |
| 475 | + "execution_count": 9, |
498 | 476 | "metadata": {}, |
499 | 477 | "outputs": [ |
500 | 478 | { |
|
523 | 501 | }, |
524 | 502 | { |
525 | 503 | "cell_type": "code", |
526 | | - "execution_count": null, |
| 504 | + "execution_count": 10, |
527 | 505 | "metadata": {}, |
528 | 506 | "outputs": [], |
529 | 507 | "source": [ |
|
534 | 512 | }, |
535 | 513 | { |
536 | 514 | "cell_type": "code", |
537 | | - "execution_count": null, |
| 515 | + "execution_count": 11, |
538 | 516 | "metadata": {}, |
539 | 517 | "outputs": [], |
540 | 518 | "source": [ |
|
543 | 521 | }, |
544 | 522 | { |
545 | 523 | "cell_type": "code", |
546 | | - "execution_count": null, |
| 524 | + "execution_count": 12, |
547 | 525 | "metadata": {}, |
548 | 526 | "outputs": [ |
549 | 527 | { |
550 | 528 | "data": { |
551 | 529 | "text/plain": [ |
552 | 530 | "MXResnet50 constructor\n", |
553 | 531 | " c_in: 3, c_out: 10\n", |
554 | | - " expansion: 4, groups: 1, dw: False\n", |
| 532 | + " expansion: 4, groups: 1, dw: False, div_groups: None\n", |
555 | 533 | " sa: False, se: False\n", |
556 | | - " stem sizes: [3, 32, 64, 64], stide on 0\n", |
| 534 | + " stem sizes: [3, 32, 64, 64], stride on 0\n", |
557 | 535 | " body sizes [64, 128, 256, 512]\n", |
558 | 536 | " layers: [3, 4, 6, 3]" |
559 | 537 | ] |
560 | 538 | }, |
561 | | - "execution_count": null, |
| 539 | + "execution_count": 12, |
562 | 540 | "metadata": {}, |
563 | 541 | "output_type": "execute_result" |
564 | 542 | } |
|
569 | 547 | }, |
570 | 548 | { |
571 | 549 | "cell_type": "code", |
572 | | - "execution_count": null, |
| 550 | + "execution_count": 13, |
573 | 551 | "metadata": {}, |
574 | 552 | "outputs": [ |
575 | 553 | { |
|
578 | 556 | "(10, [3, 4, 6, 3])" |
579 | 557 | ] |
580 | 558 | }, |
581 | | - "execution_count": null, |
| 559 | + "execution_count": 13, |
582 | 560 | "metadata": {}, |
583 | 561 | "output_type": "execute_result" |
584 | 562 | } |
|
598 | 576 | } |
599 | 577 | ], |
600 | 578 | "metadata": { |
| 579 | + "interpreter": { |
| 580 | + "hash": "00caef409dd3eda0637dae1b4e57ec87cdad0fb75342334dd1d5d7d54b26108f" |
| 581 | + }, |
601 | 582 | "kernelspec": { |
602 | | - "display_name": "Python 3", |
| 583 | + "display_name": "Python 3.10.4 ('p310_pt111')", |
603 | 584 | "language": "python", |
604 | 585 | "name": "python3" |
| 586 | + }, |
| 587 | + "language_info": { |
| 588 | + "codemirror_mode": { |
| 589 | + "name": "ipython", |
| 590 | + "version": 3 |
| 591 | + }, |
| 592 | + "file_extension": ".py", |
| 593 | + "mimetype": "text/x-python", |
| 594 | + "name": "python", |
| 595 | + "nbconvert_exporter": "python", |
| 596 | + "pygments_lexer": "ipython3", |
| 597 | + "version": "3.10.4" |
605 | 598 | } |
606 | 599 | }, |
607 | 600 | "nbformat": 4, |
|
0 commit comments