|
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