Skip to content

Commit 4573278

Browse files
committed
update
1 parent edfee18 commit 4573278

File tree

8 files changed

+82
-106
lines changed

8 files changed

+82
-106
lines changed

doc/pub/week15/html/week15-bs.html

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -952,11 +952,7 @@ <h2 id="model-definition" class="anchor">Model definition </h2>
952952
<span style="color: #008000">self</span><span style="color: #666666">.</span>dec2 <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>ConvTranspose2d(<span style="color: #666666">64</span>, c, <span style="color: #666666">3</span>, padding<span style="color: #666666">=1</span>)
953953
<span style="color: #008000">self</span><span style="color: #666666">.</span>act <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>ReLU()
954954
<span style="color: #408080; font-style: italic"># timestep embedding to condition on t</span>
955-
<span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>Sequential(
956-
nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">1</span>, <span style="color: #666666">128</span>), <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
957-
nn<span style="color: #666666">.</span>ReLU(),
958-
nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">128</span>, <span style="color: #666666">128</span>), <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
959-
)
955+
<span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>Sequential(nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">1</span>, <span style="color: #666666">128</span>), nn<span style="color: #666666">.</span>ReLU(),nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">128</span>, <span style="color: #666666">128</span>))
960956

961957
<span style="color: #008000; font-weight: bold">def</span> <span style="color: #0000FF">forward</span>(<span style="color: #008000">self</span>, x, t):
962958
<span style="color: #408080; font-style: italic"># x: [B, C, H, W], t: [B]</span>
@@ -965,7 +961,7 @@ <h2 id="model-definition" class="anchor">Model definition </h2>
965961
<span style="color: #408080; font-style: italic"># add time embedding</span>
966962
t <span style="color: #666666">=</span> t<span style="color: #666666">.</span>unsqueeze(<span style="color: #666666">-1</span>)
967963
temb <span style="color: #666666">=</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp(t)
968-
temb <span style="color: #666666">=</span> temb<span style="color: #666666">.</span>view(<span style="color: #666666">-1</span>, <span style="color: #666666">128</span>, <span style="color: #666666">1</span>, <span style="color: #666666">1</span>) <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
964+
temb <span style="color: #666666">=</span> temb<span style="color: #666666">.</span>view(<span style="color: #666666">-1</span>, <span style="color: #666666">128</span>, <span style="color: #666666">1</span>, <span style="color: #666666">1</span>)
969965
h <span style="color: #666666">=</span> h <span style="color: #666666">+</span> temb
970966
h <span style="color: #666666">=</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>act(<span style="color: #008000">self</span><span style="color: #666666">.</span>dec1(h))
971967
<span style="color: #008000; font-weight: bold">return</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>dec2(h)

doc/pub/week15/html/week15-reveal.html

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -919,11 +919,7 @@ <h2 id="model-definition">Model definition </h2>
919919
<span style="color: #658b00">self</span>.dec2 = nn.ConvTranspose2d(<span style="color: #B452CD">64</span>, c, <span style="color: #B452CD">3</span>, padding=<span style="color: #B452CD">1</span>)
920920
<span style="color: #658b00">self</span>.act = nn.ReLU()
921921
<span style="color: #228B22"># timestep embedding to condition on t</span>
922-
<span style="color: #658b00">self</span>.time_mlp = nn.Sequential(
923-
nn.Linear(<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>), <span style="color: #228B22"># Changed from 64 to 128</span>
924-
nn.ReLU(),
925-
nn.Linear(<span style="color: #B452CD">128</span>, <span style="color: #B452CD">128</span>), <span style="color: #228B22"># Changed from 64 to 128</span>
926-
)
922+
<span style="color: #658b00">self</span>.time_mlp = nn.Sequential(nn.Linear(<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>), nn.ReLU(),nn.Linear(<span style="color: #B452CD">128</span>, <span style="color: #B452CD">128</span>))
927923

928924
<span style="color: #8B008B; font-weight: bold">def</span> <span style="color: #008b45">forward</span>(<span style="color: #658b00">self</span>, x, t):
929925
<span style="color: #228B22"># x: [B, C, H, W], t: [B]</span>
@@ -932,7 +928,7 @@ <h2 id="model-definition">Model definition </h2>
932928
<span style="color: #228B22"># add time embedding</span>
933929
t = t.unsqueeze(-<span style="color: #B452CD">1</span>)
934930
temb = <span style="color: #658b00">self</span>.time_mlp(t)
935-
temb = temb.view(-<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>, <span style="color: #B452CD">1</span>, <span style="color: #B452CD">1</span>) <span style="color: #228B22"># Changed from 64 to 128</span>
931+
temb = temb.view(-<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>, <span style="color: #B452CD">1</span>, <span style="color: #B452CD">1</span>)
936932
h = h + temb
937933
h = <span style="color: #658b00">self</span>.act(<span style="color: #658b00">self</span>.dec1(h))
938934
<span style="color: #8B008B; font-weight: bold">return</span> <span style="color: #658b00">self</span>.dec2(h)

doc/pub/week15/html/week15-solarized.html

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -898,11 +898,7 @@ <h2 id="model-definition">Model definition </h2>
898898
<span style="color: #658b00">self</span>.dec2 = nn.ConvTranspose2d(<span style="color: #B452CD">64</span>, c, <span style="color: #B452CD">3</span>, padding=<span style="color: #B452CD">1</span>)
899899
<span style="color: #658b00">self</span>.act = nn.ReLU()
900900
<span style="color: #228B22"># timestep embedding to condition on t</span>
901-
<span style="color: #658b00">self</span>.time_mlp = nn.Sequential(
902-
nn.Linear(<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>), <span style="color: #228B22"># Changed from 64 to 128</span>
903-
nn.ReLU(),
904-
nn.Linear(<span style="color: #B452CD">128</span>, <span style="color: #B452CD">128</span>), <span style="color: #228B22"># Changed from 64 to 128</span>
905-
)
901+
<span style="color: #658b00">self</span>.time_mlp = nn.Sequential(nn.Linear(<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>), nn.ReLU(),nn.Linear(<span style="color: #B452CD">128</span>, <span style="color: #B452CD">128</span>))
906902

907903
<span style="color: #8B008B; font-weight: bold">def</span> <span style="color: #008b45">forward</span>(<span style="color: #658b00">self</span>, x, t):
908904
<span style="color: #228B22"># x: [B, C, H, W], t: [B]</span>
@@ -911,7 +907,7 @@ <h2 id="model-definition">Model definition </h2>
911907
<span style="color: #228B22"># add time embedding</span>
912908
t = t.unsqueeze(-<span style="color: #B452CD">1</span>)
913909
temb = <span style="color: #658b00">self</span>.time_mlp(t)
914-
temb = temb.view(-<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>, <span style="color: #B452CD">1</span>, <span style="color: #B452CD">1</span>) <span style="color: #228B22"># Changed from 64 to 128</span>
910+
temb = temb.view(-<span style="color: #B452CD">1</span>, <span style="color: #B452CD">128</span>, <span style="color: #B452CD">1</span>, <span style="color: #B452CD">1</span>)
915911
h = h + temb
916912
h = <span style="color: #658b00">self</span>.act(<span style="color: #658b00">self</span>.dec1(h))
917913
<span style="color: #8B008B; font-weight: bold">return</span> <span style="color: #658b00">self</span>.dec2(h)

doc/pub/week15/html/week15.html

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -975,11 +975,7 @@ <h2 id="model-definition">Model definition </h2>
975975
<span style="color: #008000">self</span><span style="color: #666666">.</span>dec2 <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>ConvTranspose2d(<span style="color: #666666">64</span>, c, <span style="color: #666666">3</span>, padding<span style="color: #666666">=1</span>)
976976
<span style="color: #008000">self</span><span style="color: #666666">.</span>act <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>ReLU()
977977
<span style="color: #408080; font-style: italic"># timestep embedding to condition on t</span>
978-
<span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>Sequential(
979-
nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">1</span>, <span style="color: #666666">128</span>), <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
980-
nn<span style="color: #666666">.</span>ReLU(),
981-
nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">128</span>, <span style="color: #666666">128</span>), <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
982-
)
978+
<span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp <span style="color: #666666">=</span> nn<span style="color: #666666">.</span>Sequential(nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">1</span>, <span style="color: #666666">128</span>), nn<span style="color: #666666">.</span>ReLU(),nn<span style="color: #666666">.</span>Linear(<span style="color: #666666">128</span>, <span style="color: #666666">128</span>))
983979

984980
<span style="color: #008000; font-weight: bold">def</span> <span style="color: #0000FF">forward</span>(<span style="color: #008000">self</span>, x, t):
985981
<span style="color: #408080; font-style: italic"># x: [B, C, H, W], t: [B]</span>
@@ -988,7 +984,7 @@ <h2 id="model-definition">Model definition </h2>
988984
<span style="color: #408080; font-style: italic"># add time embedding</span>
989985
t <span style="color: #666666">=</span> t<span style="color: #666666">.</span>unsqueeze(<span style="color: #666666">-1</span>)
990986
temb <span style="color: #666666">=</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>time_mlp(t)
991-
temb <span style="color: #666666">=</span> temb<span style="color: #666666">.</span>view(<span style="color: #666666">-1</span>, <span style="color: #666666">128</span>, <span style="color: #666666">1</span>, <span style="color: #666666">1</span>) <span style="color: #408080; font-style: italic"># Changed from 64 to 128</span>
987+
temb <span style="color: #666666">=</span> temb<span style="color: #666666">.</span>view(<span style="color: #666666">-1</span>, <span style="color: #666666">128</span>, <span style="color: #666666">1</span>, <span style="color: #666666">1</span>)
992988
h <span style="color: #666666">=</span> h <span style="color: #666666">+</span> temb
993989
h <span style="color: #666666">=</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>act(<span style="color: #008000">self</span><span style="color: #666666">.</span>dec1(h))
994990
<span style="color: #008000; font-weight: bold">return</span> <span style="color: #008000">self</span><span style="color: #666666">.</span>dec2(h)
0 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)