lgvaz / faststyle

Fast style transfer
Apache License 2.0
21 stars 0 forks source link

tfms on LayerFeats is confusing #6

Open lgvaz opened 4 years ago

lgvaz commented 4 years ago

The tfms argument is only used for the NormalizeAll transform and should not receive other transforms.

Can possibly substitute it directly with pretrained_img_stats or something like that.