Timothy Kassis

Remove the last '1' argument.

Yes, I can help with that. I'll put together an example notebook.

Oh wow! Great work! I totally forgot about this notebook by the way!

I've been having an issue with this as well and can't really explain it. I was able to get predictions using the following: ```python test_df = pd.read_pickle('test_path') X, y =...

I have a related suggestion. I'd like to be able to include a skipped image in the export. A few object detection models I'm using accept negative images (i.e. images...

Try: `conda install pytorch torchvision cudatoolkit=10.2 -c pytorch`

Before that make sure conda is up to date by running: `conda update conda`

Any chance someone figured out why fp16 is not working?

@lucidrains added Pytorch's Amp to his Lightweight-GAN repo and it works great on my Titan RTX!

> @TKassis Thats very useful info! Do you find lightweight gan to work better for you? It trains much much faster, but I haven't compared the two on the same...