3 Commits

Author	SHA1	Message	Date
Laurent Mazare	74bf6994b1	Move the image tensor to the appropriate device. (#1856)	2024-03-16 22:25:46 +01:00
Laurent Mazare	55bc3382cf	Allow for different behavior between training and eval (#1213) ... * Forward with training. * Do not use dropout on vgg evaluation.	2023-10-29 07:53:09 +01:00
drbh	dece37c6f4	feat: implement VGG13, VGG16 and VGG19 (#1211) ... * feat: implement VGG13, VGG16 and VGG19 * Cosmetic fixes. * More cosmetic tweaks + avoid re-loading the weights on each final layer. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>	2023-10-29 06:10:23 +00:00