Ok, I downgraded to pytorch 0.2 and things are working! Almost!
I’ll post the updated code shortly, but right now I am seeing different results from the predict(isTest=true) and the predict_array() function. Pasting my results below. Do I need to perform anything further on the values to make them probabilities? When I use the test method, it gives me the following greater than 0.1
[('Cocktail', 0.14912225), ('Day', 0.5100178), ('Dresses', 0.96452457), ('Evening', 0.14578763), ('Women', 0.9774918)]
These are the results from the predict_array() (I wonder if it is because this method uses TTA?)
[('Accents', 0.00016732486), ('Athletic', 0.0032873736), ('Baby_Bags', 0.00022524725), ('Baby_Boys_(6M-4T)', 0.001174089), ('Baby_Diaper_Bags', 0.000107968815), ('Baby_Gear', 0.00016105147), ('Baby_Girls_(6M-4T)', 0.0019865911), ('Baby_HealthAndSafety', 0.00010623256), ('Baby_Soothers', 7.480164e-05), ('Backpacks', 0.00078460766), ('Bags', 0.00064288586), ('Bath', 9.486178e-05), ('Bath_&_Body', 5.9898153e-05), ('BedAndBath', 3.510846e-05), ('Bedding', 0.00019775194), ('Bedroom', 0.00013145003), ('Big_&_Tall', 0.00080732664), ('Black', 0.0006786501), ('Blankets', 0.00010738172), ('Blazers', 0.0020364318), ('Blazers_&_Sport_Coats', 0.00015124948), ('Books_for_Mom_&_Dad', 0.00011018093), ('Bootcut', 0.0003938796), ('Boots', 0.0019993237), ('Boxers', 0.00037160324), ('Boys', 0.00036906605), ('Boys_(4-14)', 0.0025176865), ('Bracelets', 0.00019608765), ('Bras', 0.0015628829), ('Bridal', 0.0006069926), ('Bride', 0.00020638724), ('Bridesmaids', 0.0006816471), ('Briefs', 0.00033112863), ('Button_Front', 0.0007024863), ('Camisoles', 0.00036296254), ('Candles_&_Candleholders', 0.00015723538), ('Cardigans', 0.0020587104), ('Cardigans_&_Zips', 0.00022599609), ('Cargo', 0.00018141429), ('Cashmere', 0.0018986912), ('Casual', 0.010617326), ('Chemises', 0.00034301408), ('Chinos_&_Khakis', 0.00036404442), ('Classic', 0.00022426918), ('Clutches', 0.0006298735), ('Coats', 0.0009150491), ('Coats_&_Jackets', 0.0001359221), ('Cocktail', 0.0010065266), ('Cover-ups', 0.0005650788), ('Crewnecks_&_Scoopnecks', 0.0014521898), ('Cropped', 0.0016032938), ('Cuff_links', 0.00025777702), ('Day', 0.0017344432), ('Denim', 5.1488005e-05), ('Desktop', 0.00016285453), ('Diamonds', 0.0001430277), ('Diapering', 8.485448e-05), ('Distressed', 0.00021082595), ('Dress', 0.0021028225), ('Dresses', 0.0022159112), ('Dresses_&_Skirts', 0.0005268727), ('Earrings', 0.000108607375), ('Easy_fit', 0.00015920658), ('Evening', 0.0021729378), ('Flare', 0.00048548356), ('Flats', 0.0014720984), ('Flip_Flops', 0.00018539395), ('Formal', 0.00023766384), ('Frames', 0.00014337152), ('Fur_&_Shearling', 4.4151748e-05), ('Girls', 0.00082471396), ('Girls_(4-14)', 0.0031482964), ('Gowns', 0.00013934984), ('Grooming', 9.887882e-05), ('Halters', 0.00048710156), ('Handbags', 0.004061005), ('Hanging', 9.116976e-05), ('Hobos', 0.00057393), ('Hoodies', 0.0010208465), ('Hoodies_&_Sweats', 0.0006000755), ('Hosiery', 0.00021154399), ('Intimates', 0.0013576981), ('Jackets', 0.008256719), ('Jackets_&_Blazers', 0.0003360241), ('Jeans', 0.0038771238), ('Kids_&_Baby', 0.009417935), ('Kids_&_Baby_Clothing', 0.0010321799), ('Kids_BooksMusicAndDVDs', 4.443564e-05), ('Kids_Nursery', 0.00014779817), ('Knits', 0.00010930567), ('Lace-up', 5.7723e-05), ('Leather_&_Suede', 0.00035607183), ('Leggings', 0.00014404666), ('Living', 0.00015054875), ('Loafers_&_Oxfords', 9.3932016e-05), ('Long', 0.00054531614), ('Longsleeve', 0.005319159), ('Loose', 0.00016051215), ('Low_rise', 0.00045352193), ('Luggage', 0.0001760431), ('Maternity', 0.00028824923), ('Men', 0.061235134), ('Mid-length', 0.001088919), ('Mini', 0.0017485631), ('Mules_&_Clogs', 0.0011836574), ('One-Piece', 0.0001641961), ('One_Piece', 0.000108851964), ('Outerwear', 0.002972885), ('Overcoats_&_Trenchcoats', 0.00012046985), ('Pajamas', 0.0006723163), ('Panties', 0.00021483365), ('Panties_&_Briefs', 0.00013675896), ('Pants', 0.0036383008), ('Petites', 0.005360987), ('Platforms', 0.001284006), ('Plus_Sizes', 0.0030251932), ('Polo', 0.0001311797), ('Polos', 0.00077948533), ('Puffers', 0.00015336105), ('Pumps', 0.0026335984), ('Raincoats_&_Trenchcoats', 0.00041060834), ('Relaxed', 0.00026290197), ('Rings', 7.5471056e-05), ('Robes', 0.00028206446), ('Sandals', 0.0035657517), ('Satchels', 0.0013439724), ('Shapewear', 0.00026807768), ('Shirts', 0.004053797), ('Shirts_&_Blouses', 0.00019136177), ('Shoes', 0.092359), ('Shorts', 0.001934331), ('Shorts_&_Swimwear', 0.00042759473), ('Shortsleeve', 0.0036094908), ('Shoulder', 0.0027915984), ('Skinny', 0.0007575918), ('Skirts', 0.0024710628), ('Skirts_&_Skorts', 0.00018161486), ('Sleepwear', 0.0002171596), ('Sleeveless', 0.0016596871), ('Slip-ons_&_Loafers', 6.4187436e-05), ('Slippers', 0.000106401065), ('Sneakers', 8.0541555e-05), ('Socks', 0.0009950944), ('Straight', 0.00060228235), ('Straight_leg', 0.00064536225), ('Stretch', 0.00087790185), ('Suits', 0.0007665532), ('Sweat', 4.0335337e-05), ('Sweaters', 0.0070142723), ('Swimwear', 0.0028410123), ('Tanks', 0.0017036005), ('Teens', 0.0096363), ('Tees', 0.0011511133), ('Thongs', 0.00018319149), ('Tops', 0.010634665), ('Totes', 0.0026202009), ('Toys', 6.562565e-05), ('Toys_&_Gifts', 0.0005649889), ('Turtlenecks', 0.00039182324), ('Two-Piece', 0.00052589754), ('Undershirts', 0.00014653995), ('Underwear', 0.0001533128), ('Underwear_&_Socks', 0.0007373221), ('Unisex', 0.00010759788), ('V-Necks', 0.0007169939), ('Vases', 0.0001314805), ('Vests', 0.00048777252), ('Wallets', 0.00013297345), ('Watches', 0.00036453916), ('Watches_&_Jewelry', 0.00046167715), ('Wedges', 0.0015588974), ('Wide_Leg', 0.0007479941), ('Women', 0.8525294), ('Wool', 0.00011860948), ('Young_Men', 0.0016792694), ('Young_Women', 0.010041295)]