Commit ed639463 authored by Julian Johannes Steinsberger-Dührßen's avatar Julian Johannes Steinsberger-Dührßen
Browse files

Finished LayerPathParameter tagging

parent 2c7fb9e5
Pipeline #319035 failed with stage
in 40 seconds
......@@ -449,8 +449,8 @@ class ${tc.fileNameWithoutEnding}:
with autograd.record():
<#include "pythonExecuteTrain.ftl">
losses = [0]*num_pus
for i in range(num_pus):
losses = [0]*num_pus
for element in lossList[i]:
losses[i] = losses[i] + element
......
......@@ -17,7 +17,7 @@
self.${element.name} = EpisodicMemory(replay_interval=${replayInterval}, replay_batch_size=${replayBatchSize}, replay_steps=${replaySteps},
replay_gradient_steps=${replayGradientSteps}, store_prob=${replayMemoryStoreProb},
max_stored_samples=${maxStoredSamples}, use_replay=${useReplay},
query_net_dir="${queryNetDir}",
query_net_dir="${queryNetDir}/",
query_net_prefix="${queryNetPrefix}",
query_net_num_inputs=${queryNetNumInputs})
<#elseif mode == "FORWARD_FUNCTION">
......
......@@ -29,7 +29,7 @@
</#if>
with warnings.catch_warnings():
warnings.simplefilter("ignore")
self.${element.name} = gluon.nn.SymbolBlock.imports("${networkDir}" + symbolFile, inputNames, "${networkDir}" + weightFile, ctx=mx_context)
self.${element.name} = gluon.nn.SymbolBlock.imports("${networkDir}/" + symbolFile, inputNames, "${networkDir}/" + weightFile, ctx=mx_context)
self.${element.name}out_shape = self.${element.name}(*zeroInputs).shape
if self.${element.name}out_shape != ${outputShape}:
outputSize=1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment