VAE Encode Batched ๐ŸŽฅ๐Ÿ…ฅ๐Ÿ…—๐Ÿ…ข

่Š‚็‚นๅŠŸ่ƒฝ๏ผš็”จไบŽๅฐ†ๅ›พๅƒ็ผ–็ ไธบๆฝœ็ฉบ้—ด๏ผˆlatent๏ผ‰่กจ็คบ๏ผŒไฝ†ไธŽๆ ‡ๅ‡† VAE Encode ่Š‚็‚นไธๅŒ็š„ๆ˜ฏ๏ผŒๅฎƒๅ…่ฎธๆ‚จๆ‰‹ๅŠจๆŒ‡ๅฎšๆ‰นๅค„็†ๅคงๅฐ๏ผŒไปŽ่€Œๆ›ดๆœ‰ๆ•ˆๅœฐ็ฎก็† VRAM ไฝฟ็”จใ€‚

่พ“ๅ…ฅๅ‚ๆ•ฐ

ๅ‚ๆ•ฐๅ็งฐ่ฏดๆ˜Ž
pixels่พ“ๅ…ฅ็š„ๅ›พๅƒๆ•ฐๆฎ๏ผŒ้€šๅธธไธบไธ€ไธชๅผ ้‡๏ผŒๅŒ…ๅซๅคšไธชๅ›พๅƒใ€‚
vaeไฝฟ็”จ็š„ VAE ๆจกๅž‹๏ผŒ็”จไบŽๅ›พๅƒ็ผ–็ ใ€‚

่พ“ๅ‡บๅ‚ๆ•ฐ

ๅ‚ๆ•ฐๅ็งฐ่ฏดๆ˜Ž
LATENT็ผ–็ ๅŽ็š„ๆฝœๅœจๅ‘้‡๏ผŒๅฝข็Šถไธบ (batch_size, latent_dim)ใ€‚

ๆŽงไปถๅ‚ๆ•ฐ

ๅ‚ๆ•ฐๅ็งฐ่ฏดๆ˜Ž
per_batchๆฏๆฌกๅค„็†็š„ๅ›พๅƒๆ•ฐ้‡ใ€‚

้€šๅธธ่ฟ™ไธ‰ไธช่Š‚็‚นๆ˜ฏๆญ้…ไฝฟ็”จ็š„ใ€‚่ง†้ข‘ๆ€ปๅธงๆ•ฐไธบ80ๅธง๏ผŒ่ฟ™ไธ‰ไธช่Š‚็‚นไธญ็š„per_batchไธบ16๏ผŒๅˆ™่ฏฅๅทฅไฝœๆต้œ€่ฆ่ฟ่กŒ5ๆฌกใ€‚ไฝ†ๆ˜ฏ๏ผŒไพ้ ่ฟ™็งๅฝขๅผ็”Ÿๆˆ่ง†้ข‘ๆ˜ฏไธๅฏๅ–็š„๏ผŒๅ› ไธบ่ง†้ข‘็š„ๅธงไธŽๅธงไน‹้—ดๆฒกๆœ‰่ฟž่ดฏๆ€ง๏ผŒๆฏไธ€ๅธงไน‹้—ด็š„ๅ›พๅƒๅทฎๅผ‚ๅ˜ๅŒ–ๅคชๅคงใ€‚