From: Francois Fleuret Date: Wed, 18 Dec 2019 16:24:04 +0000 (+0100) Subject: OCD update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=7816075cd507d14ef09df13afbc4be525bedd08c;p=pytorch.git OCD update. --- diff --git a/denoising-ae-field.py b/denoising-ae-field.py index 67d2415..cc4af81 100755 --- a/denoising-ae-field.py +++ b/denoising-ae-field.py @@ -13,7 +13,7 @@ model = nn.Sequential( nn.Linear(100, 2) ) -############################################################ +###################################################################### def data_zigzag(nb): a = torch.empty(nb).uniform_(0, 1).view(-1, 1) @@ -80,12 +80,17 @@ ax.set_xlim(-1.6, 1.6) ax.set_ylim(-1.6, 1.6) ax.set_aspect(1) -plot_field = ax.quiver(origins[:, 0].numpy(), origins[:, 1].numpy(), - field[:, 0].numpy(), field[:, 1].numpy(), - units = 'xy', scale = 1, - width = 3e-3, headwidth = 25, headlength = 25) +plot_field = ax.quiver( + origins[:, 0].numpy(), origins[:, 1].numpy(), + field[:, 0].numpy(), field[:, 1].numpy(), + units = 'xy', scale = 1, + width = 3e-3, headwidth = 25, headlength = 25 +) -plot_data = ax.scatter(data[:, 0].numpy(), data[:, 1].numpy(), s = 1, color = 'tab:blue') +plot_data = ax.scatter( + data[:, 0].numpy(), data[:, 1].numpy(), + s = 1, color = 'tab:blue' +) fig.savefig('denoising_field.pdf', bbox_inches='tight')