Commit Graph

25 Commits

Author SHA1 Message Date
Phil Wang
7a214d7109 allow for training on different image sizes, provided images are smaller than what was passed as image_size keyword on init 0.2.6 2020-10-25 13:17:42 -07:00
Phil Wang
6d1df1a970 more efficient 0.2.5 2020-10-22 22:37:06 -07:00
Phil Wang
d65a8c17a5 remove dropout from last linear to logits 0.2.4 2020-10-16 13:58:23 -07:00
Phil Wang
f7c164d910 assert minimum number of patches 0.2.3 2020-10-16 12:19:50 -07:00
Phil Wang
c7b74e0bc3 rename ipy notebook 2020-10-14 10:35:46 -07:00
Phil Wang
5b5d98a3a7 dropouts are more specific and aggressive in the paper, thanks for letting me know @hila-chefer 2020-10-14 09:22:16 -07:00
Phil Wang
b0e4790c24 bump package 0.2.1 2020-10-13 13:12:19 -07:00
Phil Wang
0b2b3fc20c add dropouts 2020-10-13 13:11:59 -07:00
Phil Wang
ced464dcb4 Update setup.py 2020-10-11 00:06:26 -07:00
Phil Wang
5bf45a2d4d Merge pull request #4 from adimyth/main
Image Classification Example
2020-10-10 19:12:31 -07:00
adimyth
fa32e22855 adds a classification example using 'cats & dogs' data 2020-10-11 03:15:19 +05:30
Phil Wang
a0fa41070f norm cls token before sending to mlp head 0.2.0 2020-10-10 12:08:42 -07:00
Phil Wang
b298031c17 write up example for using efficient transformers 2020-10-07 19:15:21 -07:00
Phil Wang
d66b29e4cf cleanup stray print 0.0.5 2020-10-07 11:22:45 -07:00
Phil Wang
f7123720c3 add masking 0.0.4 2020-10-07 11:21:03 -07:00
Phil Wang
f5fffd9e2e remove extraneous line 2020-10-04 15:22:26 -07:00
Phil Wang
8fb261ca66 fix a bug and add suggestion for BYOL pre-training 0.0.3 2020-10-04 14:55:29 -07:00
Phil Wang
112ba5c476 update with link to Yannics video 2020-10-04 13:53:47 -07:00
Phil Wang
f899226d4f add diagram 2020-10-04 12:47:08 -07:00
Phil Wang
825a9484d1 small bug fix 0.0.2 2020-10-04 12:39:51 -07:00
Phil Wang
ee8088b3ea first commit 0.0.1 2020-10-04 12:35:01 -07:00
Phil Wang
070469d0ac Create python-publish.yml 2020-10-04 12:34:30 -07:00
Phil Wang
ea03db32f0 Update README.md 2020-10-03 15:49:27 -07:00
Phil Wang
30362d50dc Update README.md 2020-10-03 15:49:02 -07:00
Phil Wang
efb40e0b01 Initial commit 2020-10-03 15:47:26 -07:00